ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-04-10 20:29:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed binding

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-02-29 15:24:56 +01:00
parent f0156ade4c
commit 712a7544e6
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lollms/binding.py
View File

@ -148,11 +148,14 @@ class LLMBinding:
folder_path.mkdir(parents=True, exist_ok=True)
progress_bar = tqdm(total=100, unit="%", unit_scale=True, desc=f"Downloading {url.split('/')[-1]}")
# Define callback function for urlretrieve
downloaded_size = [0]
def report_progress(block_num, block_size, total_size):
progress_bar.update(block_size/total_size)
downloaded_size[0] += block_size
callback(downloaded_size[0], total_size)
# Download file from URL to folder
try:
request.urlretrieve(url, folder_path / url.split("/")[-1], reporthook=report_progress if callback is None else callback)
request.urlretrieve(url, folder_path / url.split("/")[-1], reporthook=report_progress)
print("File downloaded successfully!")
except Exception as e:
print("Error downloading file:", e)