Spaces:
Running
Running
badaoui
commited on
Commit
·
c9dede9
1
Parent(s):
a4a38a2
fix
Browse files- optimum_neuron_export.py +2 -2
optimum_neuron_export.py
CHANGED
@@ -173,13 +173,13 @@ Neuron-optimized models can achieve high-performance inference on AWS Inferentia
|
|
173 |
if os.path.exists(folder):
|
174 |
shutil.rmtree(folder, ignore_errors=True)
|
175 |
print(f"Conversion failed with error: {e}") # Print the actual error
|
176 |
-
return str(e) # Return error code and message
|
177 |
|
178 |
finally:
|
179 |
# Ensure cleanup
|
180 |
if os.path.exists(folder):
|
181 |
shutil.rmtree(folder, ignore_errors=True)
|
182 |
|
183 |
-
return new_pr
|
184 |
|
185 |
|
|
|
173 |
if os.path.exists(folder):
|
174 |
shutil.rmtree(folder, ignore_errors=True)
|
175 |
print(f"Conversion failed with error: {e}") # Print the actual error
|
176 |
+
return "1", str(e) # Return error code and message
|
177 |
|
178 |
finally:
|
179 |
# Ensure cleanup
|
180 |
if os.path.exists(folder):
|
181 |
shutil.rmtree(folder, ignore_errors=True)
|
182 |
|
183 |
+
return "0", new_pr
|
184 |
|
185 |
|