Forráskód Böngészése

Fix the download script

oobabooga 3 éve
szülő
commit
dd1bed2d8b
1 módosított fájl, 4 hozzáadás és 2 törlés
  1. 4 2
      download-model.py

+ 4 - 2
download-model.py

@@ -27,9 +27,11 @@ def get_file(args):
             f.write(data)
         t.close()
 
-model = Path(argv[1])
+model = argv[1]
+if model[-1] == '/':
+    model = model[:-1]
 url = f'https://huggingface.co/{model}/tree/main'
-output_folder = Path("models") / model.name
+output_folder = Path("models") / model.split('/')[-1]
 if not output_folder.exists():
     output_folder.mkdir()