fix

huggingface · Sep 19, 2023 · 33957af · 33957af
1 parent 46b26b5
commit 33957af
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/optimum/onnxruntime/modeling_decoder.py b/optimum/onnxruntime/modeling_decoder.py
@@ -1023,7 +1023,8 @@ def _from_pretrained(
             except EntryNotFoundError:
                 # model doesn't use external data
                 pass
-            new_model_save_dir = Path(model_cache_path).parent
+            model_cache_path = Path(model_cache_path)
+            new_model_save_dir = model_cache_path.parent
             preprocessors = maybe_load_preprocessors(model_id, subfolder=subfolder)
 
         # model_save_dir can be provided in kwargs as a TemporaryDirectory instance, in which case we want to keep it