Update convert_voxtral_hf_to_mistral.py

convert_voxtral_hf_to_mistral.py

@@ -128,12 +128,15 @@ def write_model(
     local_path = snapshot_download(input_path_or_repo)
     
     # Convert config
-    config_path = os.path.join(local_path, "params.json")
-    with open(config_path, "r") as f:
-        config = json.load(f)
-    if os.path.exists(config_path):
-        if unquantized_model_path is not None:
-           config = add_quantization_config(config, hf_config)
+    if unquantized_model_path is not None:
+        if os.path.exists(unquantized_model_path):
+            unquantized_model_path = unquantized_model_path
+        else:
+            unquantized_model_path = snapshot_download(unquantized_model_path)
+        config_path = os.path.join(unquantized_model_path, "params.json")
+        with open(config_path, "r") as f:
+            config = json.load(f)
+        config = add_quantization_config(config, hf_config)
 
         with open(os.path.join(output_dir, "params.json"), "w") as f:
             json.dump(config, f, indent=2)