Fix the mirror Issue and update the Docs.

Author: codewithdark-git

quantllm/data/load_dataset.py

@@ -1,4 +1,4 @@
-from datasets import load_dataset, Dataset, disable_progress_bars
+from datasets import load_dataset, Dataset
 from typing import Optional, Dict, Any, Union
 import os
 from ..trainer.logger import TrainingLogger
@@ -20,7 +20,7 @@ def __init__(self, logger: Optional[TrainingLogger] = None):
             logger (TrainingLogger, optional): Logger instance
         """
         self.logger = logger or TrainingLogger()
-        disable_progress_bars()  # Disable default progress bars
+        
 
     def load_hf_dataset(
         self,

quantllm/trainer/logger.py

@@ -45,8 +45,8 @@ def log_welcome_message(self):
             f"{LogLevel.SUCCESS.value}│  💻 Supports CUDA, CPU, and Apple Silicon                              │{LogLevel.RESET.value}\n"
             f"{LogLevel.SUCCESS.value}│                                                                         │{LogLevel.RESET.value}\n"
             f"{LogLevel.SUCCESS.value}╰─────────────────────────────────────────────────────────────────────────╯{LogLevel.RESET.value}\n\n"
-        )
-        print(logo)
+            )
+            print(logo)
 
     def _format_message(self, level: LogLevel, message: str) -> str:
         """Format log message with timestamp and color."""

quantllm/utils/benchmark.py

@@ -352,7 +352,7 @@ def plot_comparison(self, save_path: str = None):
         fig.suptitle('Quantization Method Comparison')
 
         # Latency comparison
-            df['mean_latency_ms'].plot(
+        df['mean_latency_ms'].plot(
             kind='bar', ax=axes[0, 0], rot=45,
             title='Mean Inference Latency (ms)'
         )