[Fix] Fix dict update in minigpt4. (open-mmlab#1709)

ASHORE1225 · Jul 28, 2023 · 1f99279 · 1f99279
1 parent 0b96dca
commit 1f99279
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmpretrain/models/multimodal/minigpt4/minigpt4.py b/mmpretrain/models/multimodal/minigpt4/minigpt4.py
@@ -155,8 +155,8 @@ def __init__(self,
             top_p=0.9,
             repetition_penalty=1.0,
             length_penalty=1.0,
-            temperature=1.0,
-            **generation_cfg)
+            temperature=1.0)
+        self.generation_cfg.update(**generation_cfg)
 
         if hasattr(self, 'register_load_state_dict_post_hook'):
             self.register_load_state_dict_post_hook(self._load_llama_proj_hook)