fix

InternLM · Nov 5, 2024 · fea887c · fea887c
1 parent 0bfcae6
commit fea887c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lmdeploy/turbomind/deploy/target_model/base.py b/lmdeploy/turbomind/deploy/target_model/base.py
@@ -91,7 +91,7 @@ def update_model_config(self):
         final_cfg = config_to_dict(self.model_config)
         final_cfg.update(dict(start_id=bos_id, end_id=eos_id))
         final_cfg.update(self.input_model_info)
-        if not self.input_model_info['embedding_size']:
+        if 'embedding_size' not in self.input_model_info.keys():
             final_cfg.update(
                 embedding_size=self.input_model_info['vocab_size'])