diff --git a/ultralytics/nn/__init__.py b/ultralytics/nn/__init__.py
index e69de29..9889b7e 100644
--- a/ultralytics/nn/__init__.py
+++ b/ultralytics/nn/__init__.py
@@ -0,0 +1,9 @@
+# Ultralytics YOLO 🚀, AGPL-3.0 license
+
+from .tasks import (BaseModel, ClassificationModel, DetectionModel, SegmentationModel, attempt_load_one_weight,
+                    attempt_load_weights, guess_model_scale, guess_model_task, parse_model, torch_safe_load,
+                    yaml_model_load)
+
+__all__ = ('attempt_load_one_weight', 'attempt_load_weights', 'parse_model', 'yaml_model_load', 'guess_model_task',
+           'guess_model_scale', 'torch_safe_load', 'DetectionModel', 'SegmentationModel', 'ClassificationModel',
+           'BaseModel')