fix model param updates

Clarifai · Feb 6, 2025 · 14427b4 · 14427b4
1 parent 2eecf55
commit 14427b4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/clarifai/client/model.py b/clarifai/client/model.py
@@ -1233,8 +1233,9 @@ def _get_model_info_for_inference(self, inference_params: Dict = {},
 
     model_info = resources_pb2.Model()
     model_info.CopyFrom(self.model_info)
-    model_info.model_version.output_info.params.update(inference_params)
-    model_info.model_version.output_info.output_config.update(output_config)
+    model_info.model_version.output_info.params = inference_params
+    model_info.model_version.output_info.output_config.CopyFrom(
+        resources_pb2.OutputConfig(**output_config))
     return model_info
 
   def _list_concepts(self) -> List[str]: