"src/vscode:/vscode.git/clone" did not exist on "5b1af9ab8244484c5fc28a89fdbb1557106d897f"
Unverified Commit bfb118c0 authored by Minho Ryu's avatar Minho Ryu Committed by GitHub
Browse files

fix bug when eos_ids==0 (#8315)

parent f6e07f27
......@@ -475,7 +475,7 @@ class ModelConfig:
def get_hf_eos_token_id(self) -> Optional[Set[int]]:
eos_ids = getattr(self.hf_config, "eos_token_id", None)
if eos_ids:
if eos_ids is not None:
# it can be either int or list of int
eos_ids = {eos_ids} if isinstance(eos_ids, int) else set(eos_ids)
if eos_ids is None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment