Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-12-25 13:58:56 +01:00
Code Issues Packages Projects Releases 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
Show Stats 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 tqdm
import sys
from sys import argv
import argparse
from pathlib import Path
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.")
if __name__ == '__main__':
model = argv[1]
model = args.model
if model[-1] == '/':
model = model[:-1]
branch = args.branch