lazy loading, fixes, new release

src/autotrain/__init__.py

@@ -30,4 +30,4 @@
 warnings.filterwarnings("ignore", category=UserWarning, module="tensorflow")
 
 
-__version__ = "0.6.32.dev0"
+__version__ = "0.6.33.dev0"

src/autotrain/cli/run_image_classification.py

@@ -5,8 +5,6 @@
 import torch
 
 from autotrain import logger
-from autotrain.trainers.image_classification.__main__ import train as train_image_classification
-from autotrain.trainers.image_classification.params import ImageClassificationParams
 
 from . import BaseAutoTrainCommand
 
@@ -262,6 +260,9 @@ def __init__(self, args):
         self.num_gpus = torch.cuda.device_count()
 
     def run(self):
+        from autotrain.trainers.image_classification.__main__ import train as train_image_classification
+        from autotrain.trainers.image_classification.params import ImageClassificationParams
+
         logger.info("Running Text Classification")
         if self.args.train:
             params = ImageClassificationParams(

src/autotrain/cli/run_llm.py

@@ -6,10 +6,6 @@
 import torch
 
 from autotrain import logger
-from autotrain.backend import EndpointsRunner, SpaceRunner
-from autotrain.infer.text_generation import TextGenerationInference
-from autotrain.trainers.clm.__main__ import train as train_llm
-from autotrain.trainers.clm.params import LLMTrainingParams
 
 from . import BaseAutoTrainCommand
 
@@ -392,6 +388,8 @@ def __init__(self, args):
                     raise ValueError("Token must be specified for spaces backend")
 
         if self.args.inference:
+            from autotrain.infer.text_generation import TextGenerationInference
+
             tgi = TextGenerationInference(
                 self.args.project_name, use_int4=self.args.use_int4, use_int8=self.args.use_int8
             )
@@ -407,6 +405,10 @@ def __init__(self, args):
         self.num_gpus = torch.cuda.device_count()
 
     def run(self):
+        from autotrain.backend import EndpointsRunner, SpaceRunner
+        from autotrain.trainers.clm.__main__ import train as train_llm
+        from autotrain.trainers.clm.params import LLMTrainingParams
+
         logger.info("Running LLM")
         logger.info(f"Params: {self.args}")
         if self.args.train:

src/autotrain/cli/run_tabular.py

@@ -6,8 +6,6 @@
 
 from autotrain import logger
 from autotrain.backend import EndpointsRunner, SpaceRunner
-from autotrain.trainers.tabular.__main__ import train as train_tabular
-from autotrain.trainers.tabular.params import TabularParams
 
 from . import BaseAutoTrainCommand
 
@@ -229,6 +227,9 @@ def __init__(self, args):
         self.args.target_columns = [k.strip() for k in self.args.target_columns.split(",")]
 
     def run(self):
+        from autotrain.trainers.tabular.__main__ import train as train_tabular
+        from autotrain.trainers.tabular.params import TabularParams
+
         logger.info("Running Tabular Training...")
         if self.args.train:
             params = TabularParams(

src/autotrain/cli/run_text_classification.py

@@ -7,8 +7,6 @@
 
 from autotrain import logger
 from autotrain.backend import EndpointsRunner, SpaceRunner
-from autotrain.trainers.text_classification.__main__ import train as train_text_classification
-from autotrain.trainers.text_classification.params import TextClassificationParams
 
 from . import BaseAutoTrainCommand
 
@@ -294,6 +292,9 @@ def __init__(self, args):
             self.args.token = os.environ.get("HF_TOKEN", None)
 
     def run(self):
+        from autotrain.trainers.text_classification.__main__ import train as train_text_classification
+        from autotrain.trainers.text_classification.params import TextClassificationParams
+
         logger.info("Running Text Classification")
         if self.args.train:
             params = TextClassificationParams(