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

upgrade

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-02-29 18:37:51 +01:00
parent 8d48f95264
commit 1813e4505b
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lollms/binding.py
View File

@ -155,14 +155,15 @@ class LLMBinding:
downloaded_size = [0]
def report_progress(block_num, block_size, total_size):
progress_bar.update(block_size/total_size)
downloaded_size[0] += block_size
downloaded_size[0] += block_size*100
callback(downloaded_size[0], total_size)
# Download file from URL to folder
try:
Path(model_full_path).parent.mkdir(parents=True, exist_ok=True)
request.urlretrieve(url, model_full_path, reporthook=report_progress)
print("File downloaded successfully!")
except Exception as e:
print("Error downloading file:", e)
ASCIIColors.error("Error downloading file:", e)
sys.exit(1)
def reference_model(self, path):