mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Update download-model.py

Browse source
This commit is contained in:
oobabooga 2023-01-21 00:38:23 -03:00 • committed by GitHub
parent fffd49e64e
commit 18ef72d7c0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
download-model.py
View file

@ -11,7 +11,6 @@ from bs4 import BeautifulSoup
import multiprocessing import multiprocessing
import tqdm import tqdm
import sys import sys
from sys import argv
import argparse import argparse
from pathlib import Path from pathlib import Path
import re import re
@ -43,7 +42,7 @@ def sanitize_branch_name(branch_name):
raise ValueError("Invalid branch name. Only alphanumeric characters, period, underscore and dash are allowed.") raise ValueError("Invalid branch name. Only alphanumeric characters, period, underscore and dash are allowed.")
if __name__ == '__main__': if __name__ == '__main__':
model = argv[1] model = args.model
if model[-1] == '/': if model[-1] == '/':
model = model[:-1] model = model[:-1]
branch = args.branch branch = args.branch