Skip to content

Commit 1fc1880

Browse files
committed
[bugfix] bugfix of None stop_ids.
1 parent e2edef4 commit 1fc1880

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

llm_export.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -918,7 +918,7 @@ def load_model(self, model_path):
918918
elif isinstance(eos_token_id, Iterable):
919919
for id in eos_token_id:
920920
self.stop_ids.append(id)
921-
921+
self.stop_ids = [stop_id for stop_id in self.stop_ids if stop_id is not None]
922922
model_mapper = ModelMapper()
923923
self.model_type, self.model_map = model_mapper.get_map(self.config)
924924
# print(self.model)

0 commit comments

Comments
 (0)