Update optimum_neuron_export.py

optimum_neuron_export.py

@@ -330,25 +330,9 @@ def export_decoder_model(model_id: str, folder: str, token:str) -> Generator:
         model = NeuronModelForCausalLM.export(
                 model_id=model_id, neuron_config=neuron_config,
             )
+        model.save_pretrained(folder)
         
-        yield "✅ Export/compilation completed successfully."
-        
-        # Find the newly created cache artifacts
-        yield "🔍 Locating compiled artifacts in Neuron cache..."
-        cache_artifact_dir = find_neuron_cache_artifacts()
-        
-        if not cache_artifact_dir:
-            raise Exception("❌ Could not find compiled artifacts in Neuron cache")
-        
-        yield f"📂 Found artifacts at: {cache_artifact_dir}"
-        
-        # Copy artifacts from cache to our target folder
-        yield f"📋 Copying artifacts to export folder..."
-        if os.path.exists(folder):
-            shutil.rmtree(folder)
-        shutil.copytree(cache_artifact_dir, folder)
-        
-        yield f"✅ Artifacts successfully copied to {folder}"
+        yield f"✅ Decoder model export completed. Files saved to {folder}"
         
     except Exception as e:
         yield f"❌ Export failed with error: {e}"