Fixed some errors

Author: UtkarshTheDev

CHANGELOG.md

@@ -16,6 +16,16 @@ All notable changes to LocalLab will be documented in this file.
 - Added informative messages before and after model downloads for better user experience
 - Ensured consistent progress bar display across different model types and sizes
 
+## [0.6.4] - 2025-05-16
+
+### Fixed
+
+- Fixed critical error with Hugging Face progress bars display
+- Corrected function naming and imports for better compatibility
+- Improved early configuration system to properly set up logging
+- Enhanced error handling during model downloads
+- Fixed AttributeError with huggingface_hub module
+
 ## [0.6.3] - 2025-05-16
 
 ### Improved

locallab/__init__.py

@@ -4,18 +4,18 @@
 
 # Import early configuration first to set up logging and environment variables
 # This ensures Hugging Face's progress bars are displayed correctly
-from .utils.early_config import enable_hf_progress_bars
+from .utils.early_config import configure_hf_logging
 
-__version__ = "0.6.4"  # Updated to improve model downloading experience and fix CLI settings
+__version__ = "0.6.4"  # Fixed Hugging Face progress bars display and improved model downloading experience
 
 # Only import what's necessary initially, lazy-load the rest
 from .logger import get_logger
 
 # Explicitly expose start_server for direct import
 from .server import start_server, cli
 
-# Enable Hugging Face progress bars with native display
-enable_hf_progress_bars()
+# Configure Hugging Face logging early
+configure_hf_logging()
 
 # Other imports will be lazy-loaded when needed
 # from .config import MODEL_REGISTRY, DEFAULT_MODEL

locallab/model_manager.py

@@ -1,6 +1,6 @@
 # Import early configuration module first to set up logging and environment variables
 # This ensures Hugging Face's progress bars are displayed correctly
-from .utils.early_config import enable_hf_progress_bars, StdoutRedirector
+from .utils.early_config import configure_hf_progress_bars, StdoutRedirector
 
 from .config import HF_TOKEN_ENV, get_env_var, set_env_var
 import os
@@ -26,7 +26,7 @@
 
 # Enable Hugging Face progress bars with native display
 # This ensures we see the visually appealing progress bars from HuggingFace
-enable_hf_progress_bars()
+configure_hf_progress_bars()
 
 # Import transformers after configuring logging to ensure proper display
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
@@ -278,7 +278,7 @@ async def _load_model_with_optimizations(self, model_id: str):
             print(f"\n{Fore.CYAN}Starting model download - native progress bars will appear below{Style.RESET_ALL}\n")
 
             # Enable Hugging Face progress bars again to ensure they're properly configured
-            enable_hf_progress_bars()
+            configure_hf_progress_bars()
 
             # Use a context manager to ensure proper display of Hugging Face progress bars
             with StdoutRedirector(disable_logging=True):
@@ -1093,7 +1093,7 @@ async def load_custom_model(self, model_name: str, fallback_model: Optional[str]
             print(f"\n{Fore.CYAN}Starting custom model download - native progress bars will appear below{Style.RESET_ALL}\n")
 
             # Enable Hugging Face progress bars again to ensure they're properly configured
-            enable_hf_progress_bars()
+            configure_hf_progress_bars()
 
             # Use a context manager to ensure proper display of Hugging Face progress bars
             with StdoutRedirector(disable_logging=True):

locallab/utils/early_config.py

@@ -26,17 +26,17 @@ def configure_hf_logging():
     """
     # Disable all warnings
     warnings.filterwarnings("ignore")
-    
+
     # Configure logging for Hugging Face libraries
     for logger_name in ["transformers", "huggingface_hub", "accelerate", "tqdm", "filelock"]:
         hf_logger = logging.getLogger(logger_name)
         hf_logger.setLevel(logging.WARNING)  # Only show warnings and errors
         hf_logger.propagate = False  # Don't propagate to parent loggers
-        
+
         # Remove any existing handlers
         for handler in hf_logger.handlers[:]:
             hf_logger.removeHandler(handler)
-        
+
         # Add a null handler to prevent warnings about no handlers
         hf_logger.addHandler(logging.NullHandler())
 
@@ -54,17 +54,17 @@ def __init__(self, disable_logging=True):
         self.original_stdout = sys.stdout
         self.original_stderr = sys.stderr
         self.original_log_levels = {}
-    
+
     def __enter__(self):
         # Store original log levels
         if self.disable_logging:
             for logger_name in ["transformers", "huggingface_hub", "accelerate", "tqdm", "filelock"]:
                 logger = logging.getLogger(logger_name)
                 self.original_log_levels[logger_name] = logger.level
                 logger.setLevel(logging.WARNING)
-        
+
         return self
-    
+
     def __exit__(self, exc_type, exc_val, exc_tb):
         # Restore original log levels
         if self.disable_logging:
@@ -83,18 +83,30 @@ def enable_hf_progress_bars():
         tqdm.tqdm.monitor_interval = 0  # Disable monitor thread
     except ImportError:
         pass
-    
+
     # Configure huggingface_hub
     try:
         import huggingface_hub
-        huggingface_hub.enable_progress_bars()
+        # The correct way to enable progress bars in huggingface_hub
+        from huggingface_hub.utils import logging as hf_logging
+        hf_logging.enable_progress_bars()
+
+        # Also enable HF Transfer for better download experience
+        if hasattr(huggingface_hub, "constants"):
+            huggingface_hub.constants.HF_HUB_ENABLE_HF_TRANSFER = True
     except ImportError:
         pass
-    
+
     # Configure transformers
     try:
         import transformers
         transformers.utils.logging.enable_progress_bar()
         transformers.logging.set_verbosity_warning()
     except ImportError:
         pass
+
+# Alias for backward compatibility
+configure_hf_progress_bars = enable_hf_progress_bars
+
+# Export the configure_hf_logging function for use in __init__.py
+__all__ = ["enable_hf_progress_bars", "configure_hf_progress_bars", "configure_hf_logging", "StdoutRedirector"]

setup.py

@@ -59,7 +59,7 @@
     },
     author="Utkarsh Tiwari",
     author_email="utkarshweb2023@gmail.com",
-    description="LocalLab: Run language models locally or in Google Collab with a friendly API",
+    description="LocalLab: A lightweight AI inference server for running LLMs locally or in Google Colab with a friendly API.",
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/UtkarshTheDev/LocalLab",