mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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
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()