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

enhanced fix

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-02-29 20:00:54 +01:00
parent d2d039ac81
commit 4bdc42ae74
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
lollms/binding.py
View File

@ -160,13 +160,16 @@ class LLMBinding:
else:
# Create folder if it doesn't exist
folder_path.mkdir(parents=True, exist_ok=True)
progress_bar = tqdm(total=100, unit="%", unit_scale=True, desc=f"Downloading {url.split('/')[-1]}")
if not callback:
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*100
callback(downloaded_size[0], total_size)
if callback:
downloaded_size[0] += block_size
callback(downloaded_size[0], total_size)
else:
progress_bar.update(block_size/total_size)
# Download file from URL to folder
try:
Path(model_full_path).parent.mkdir(parents=True, exist_ok=True)