From c5f8b73d9dab26040123b35942e98bbe099f820e Mon Sep 17 00:00:00 2001
From: filipstrand <strand.filip@gmail.com>
Date: Wed, 3 Dec 2025 22:50:02 +0100
Subject: [PATCH] Fix import

---
 README.md             | 2 +-
 src/mflux/__init__.py | 5 -----
 2 files changed, 1 insertion(+), 6 deletions(-)

diff --git a/README.md b/README.md
index 5b6595a..55383dd 100644
--- a/README.md
+++ b/README.md
@@ -172,7 +172,7 @@ This is useful for integrating MFLUX into shell scripts or dynamically generatin
 Alternatively, you can use MFLUX directly in Python:
 
 ```python
-from mflux import Flux1
+from mflux.models.flux.variants.txt2img.flux import Flux1
 
 # Load the model
 flux = Flux1.from_name(
diff --git a/src/mflux/__init__.py b/src/mflux/__init__.py
index 9795341..c72ee76 100644
--- a/src/mflux/__init__.py
+++ b/src/mflux/__init__.py
@@ -4,8 +4,3 @@ import os
 # This must be set before any tokenizers are imported/used
 if "TOKENIZERS_PARALLELISM" not in os.environ:
     os.environ["TOKENIZERS_PARALLELISM"] = "false"
-
-# Export main classes for convenient import
-from mflux.models.flux.variants.txt2img.flux import Flux1
-
-__all__ = ["Flux1"]