From 712a7544e669ecc9d869f2d90a510541f47597e6 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Thu, 29 Feb 2024 15:24:56 +0100 Subject: [PATCH] fixed binding --- lollms/binding.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/lollms/binding.py b/lollms/binding.py index 902527b..f41b120 100644 --- a/lollms/binding.py +++ b/lollms/binding.py @@ -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)