zhengjl/text-generation-webui
Public Access
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix the download script (sort of)

Browse Source
This commit is contained in:
oobabooga
2023-03-02 14:05:21 -03:00
parent bcea196c9d
commit 7c70e0e2a6
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
download-model.py
View File

@@ -102,8 +102,8 @@ def get_download_links_from_huggingface(model, branch):
content = requests.get(f"{base}{page}").content content = requests.get(f"{base}{page}").content
dict = json.loads(content) dict = json.loads(content)
for i in range(len(dict['items'])): for i in range(len(dict)):
fname = dict['items'][i]['path'] fname = dict[i]['path']
is_pytorch = re.match("pytorch_model.*\.bin", fname) is_pytorch = re.match("pytorch_model.*\.bin", fname)
is_safetensors = re.match("model.*\.safetensors", fname) is_safetensors = re.match("model.*\.safetensors", fname)
@@ -123,7 +123,8 @@ def get_download_links_from_huggingface(model, branch):
has_pytorch = True has_pytorch = True
classifications.append('pytorch') classifications.append('pytorch')
page = dict['nextUrl'] #page = dict['nextUrl']
page = None
# If both pytorch and safetensors are available, download safetensors only # If both pytorch and safetensors are available, download safetensors only
if has_pytorch and has_safetensors: if has_pytorch and has_safetensors: