Fix Vision Load checkpoints for Gemma-3-4b-it

Author: MohammedTaherMcW

models/tt_transformers/tt/model_config.py

@@ -1679,9 +1679,10 @@ def merge_vision_config(base_config):
 
             if "text_config" in self.hf_config or "vision_config" in self.hf_config:
                 if "gemma-3-4b" in self.base_model_name:
-                    merged_text_config = merge_text_config(self.hf_config)
-                    self._set_params_from_dict(merged_text_config, is_hf=True)
-                    self._set_vision_params(self.hf_config)
+                    self._set_params_from_dict(self.hf_config, is_hf=True)
+                    if "vision_config" in self.hf_config:
+                        merged_vision_config = merge_vision_config(self.hf_config)
+                        self._set_vision_params(merged_vision_config)
                 else:
                     merged_text_config = merge_text_config(self.hf_config)
                     self._set_params_from_dict(merged_text_config, is_hf=True)