Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-25 17:29:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix the download script

Browse Source
This commit is contained in:
oobabooga 2023-01-07 16:49:21 -03:00
parent 5345685ead
commit dd1bed2d8b
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
download-model.py
View File

@ -27,9 +27,11 @@ def get_file(args):
f.write(data) f.write(data)
t.close() t.close()
model = Path(argv[1]) model = argv[1]
if model[-1] == '/':
model = model[:-1]
url = f'https://huggingface.co/{model}/tree/main' 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(): if not output_folder.exists():
output_folder.mkdir() output_folder.mkdir()