Skip to content

Commit

Permalink
revert original context len
Browse files Browse the repository at this point in the history
  • Loading branch information
ajindal1 committed Jan 8, 2025
1 parent f644e98 commit 82fd6e0
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion src/python/py/models/builder.py
Original file line number Diff line number Diff line change
Expand Up @@ -347,7 +347,6 @@ def make_genai_config(self, model_name_or_path, extra_kwargs, out_dir):
"num_key_value_heads": self.num_kv_heads,
},
"eos_token_id": config.eos_token_id,
"original_context_length": self.original_context_length,
"pad_token_id": config.pad_token_id if hasattr(config, "pad_token_id") and config.pad_token_id is not None else config.eos_token_id[0] if isinstance(config.eos_token_id, list) else config.eos_token_id,
"type": self.model_type[ : self.model_type.find("For")].lower(),
"vocab_size": self.vocab_size,
Expand Down

0 comments on commit 82fd6e0

Please sign in to comment.