diff --git a/README.md b/README.md
index 06bebbe..f4b057e 100644
--- a/README.md
+++ b/README.md
@@ -20,7 +20,7 @@ like [Numpy](https://numpy.org) and [Pillow](https://pypi.org/project/pillow/) f
 ### Models
 
 - [x] FLUX.1-Scnhell
-- [ ] FLUX.1-Dev
+- [x] FLUX.1-Dev
 
 ### Installation
 1. Clone the repo:
diff --git a/main.py b/main.py
index 5cf39bb..8b26c5b 100644
--- a/main.py
+++ b/main.py
@@ -4,10 +4,10 @@ import sys
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), 'src')))
 
 from flux_1_schnell.config.config import Config
-from flux_1_schnell.flux import Flux1Schnell
+from flux_1_schnell.flux import Flux1
 from flux_1_schnell.post_processing.image_util import ImageUtil
 
-flux = Flux1Schnell("black-forest-labs/FLUX.1-schnell")
+flux = Flux1("black-forest-labs/FLUX.1-schnell", max_sequence_length=256)
 
 image = flux.generate_image(
     seed=3,