Add utils package init

app/utils/__init__.py

@@ -0,0 +1,59 @@
+"""
+Utilities Package
+
+Helper functions for training, datasets, models, and PEFT/LoRA.
+"""
+
+from app.utils.training_utils import (
+    get_model_class_for_task,
+    estimate_memory_requirements,
+    get_training_args,
+    get_peft_config,
+    get_data_collator,
+    compute_metrics_factory,
+    save_training_artifacts,
+    generate_job_id
+)
+from app.utils.dataset_utils import (
+    load_dataset,
+    validate_dataset,
+    tokenize_dataset,
+    prepare_dataset_for_task
+)
+from app.utils.model_utils import (
+    load_model,
+    load_tokenizer,
+    apply_peft,
+    estimate_parameters
+)
+from app.utils.peft_utils import (
+    configure_lora,
+    configure_adalora,
+    apply_peft_config
+)
+
+__all__ = [
+    # Training utils
+    "get_model_class_for_task",
+    "estimate_memory_requirements",
+    "get_training_args",
+    "get_peft_config",
+    "get_data_collator",
+    "compute_metrics_factory",
+    "save_training_artifacts",
+    "generate_job_id",
+    # Dataset utils
+    "load_dataset",
+    "validate_dataset",
+    "tokenize_dataset",
+    "prepare_dataset_for_task",
+    # Model utils
+    "load_model",
+    "load_tokenizer",
+    "apply_peft",
+    "estimate_parameters",
+    # PEFT utils
+    "configure_lora",
+    "configure_adalora",
+    "apply_peft_config"
+]