Fix exception handling in manage.py (#2912)

This commit is contained in:
Cohee authored and GitHub committed 2023-09-04 12:54:30 +02:00
1 parent 40b527345f
commit b3b1555d82
1 file changed
+1 -1
+1 -1
View File
@@ -325,7 +325,7 @@ class ModelManager(object):
elif "hf_url" in model_item:
self._download_hf_model(model_item, output_path)
except requests.Exception.RequestException as e:
except requests.RequestException as e:
print(f" > Failed to download the model file to {output_path}")
rmtree(output_path)
raise e