Update optimum_neuron_export.py

optimum_neuron_export.py

@@ -296,7 +296,7 @@ def export_transformer_model(model_id: str, task: str, folder: str, token: str)
         
         # Find the newly created cache artifacts
         yield "🔍 Locating compiled artifacts in Neuron cache..."
-        cache_artifact_dir = find_neuron_cache_artifacts(cache_base_dir)
+        cache_artifact_dir = find_neuron_cache_artifacts()
         
         if not cache_artifact_dir:
             raise Exception("❌ Could not find compiled artifacts in Neuron cache")
@@ -335,7 +335,7 @@ def export_decoder_model(model_id: str, folder: str, token:str) -> Generator:
         
         # Find the newly created cache artifacts
         yield "🔍 Locating compiled artifacts in Neuron cache..."
-        cache_artifact_dir = find_neuron_cache_artifacts(cache_base_dir)
+        cache_artifact_dir = find_neuron_cache_artifacts()
         
         if not cache_artifact_dir:
             raise Exception("❌ Could not find compiled artifacts in Neuron cache")