fixes to check_and_update.py script (#2778)

* fixes to `check_and_update.py`  script

---------

Signed-off-by: Dave Lee <dave@gray101.com>
This commit is contained in:
Dave 2024-07-12 02:40:45 -04:00 committed by GitHub
parent f5bcba70da
commit fb04347d3b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -4,7 +4,7 @@ import requests
import sys
import os
uri = sys.argv[0]
uri = sys.argv[1]
file_name = uri.split('/')[-1]
# Function to parse the URI and determine download method
@ -36,6 +36,7 @@ def manual_safety_check_hf(repo_id):
download_type, repo_id_or_url = parse_uri(uri)
new_checksum = None
file_path = None
# Decide download method based on URI type
if download_type == 'huggingface':