schnell

.gitattributes

@@ -3,3 +3,4 @@ ae.safetensors filter=lfs diff=lfs merge=lfs -text
 t5-v1_1-xxl-encoder-bf16//*.safetensor filter=lfs diff=lfs merge=lfs -text
 t5-v1_1-xxl-encoder-bf16//spiece.model filter=lfs diff=lfs merge=lfs -text
 t5-v1_1-xxl-encoder-bf16/model.safetensors filter=lfs diff=lfs merge=lfs -text
+flux1-schnell.safetensors filter=lfs diff=lfs merge=lfs -text

config.json

@@ -1,7 +1,7 @@
 {
-  "config_path": "configs/config-dev.json",
-  "flow_model_path": "flux1-dev.safetensors",
-  "model_version": "flux-dev",
+  "config_path": "configs/config-schnell.json",
+  "flow_model_path": "flux1-schnell.safetensors",
+  "model_version": "flux-schnell",
   "flux_device": "cuda",
   "autoencoder_path": null,
   "autoencoder_device": "cuda",

configs/config-schnell.json

@@ -34,16 +34,16 @@
     "scale_factor": 0.3611,
     "shift_factor": 0.1159
   },
-  "ckpt_path": "/big/generator-ui/flux-testing/flux/model-dir-schnell/flux1-schnell.sft",
-  "ae_path": "/big/generator-ui/flux-testing/flux/model-dir-schnell/ae.sft",
+  "ckpt_path": "flux1-schnell.safetensors",
+  "ae_path": "ae.safetensors",
   "repo_id": "black-forest-labs/FLUX.1-schnell",
-  "repo_flow": "flux1-schnell.sft",
-  "repo_ae": "ae.sft",
+  "repo_flow": "flux1-dev.safetensors",
+  "repo_ae": "ae.safetensors",
   "text_enc_max_length": 256,
-  "text_enc_path": "city96/t5-v1_1-xxl-encoder-bf16",
-  "text_enc_device": "cuda:1",
-  "ae_device": "cuda:1",
-  "flux_device": "cuda:0",
+  "text_enc_path": "t5-v1_1-xxl-encoder-bf16",
+  "text_enc_device": "cuda",
+  "ae_device": "cuda",
+  "flux_device": "cuda",
   "flow_dtype": "float16",
   "ae_dtype": "bfloat16",
   "text_enc_dtype": "bfloat16",
@@ -54,4 +54,4 @@
   "offload_ae": false,
   "offload_text_enc": false,
   "offload_flow": false
-}
+}

flumina.py

@@ -118,7 +118,7 @@ class Text2ImageRequest(BaseModel):
     prompt: str
     aspect_ratio: str = "16:9"
     guidance_scale: float = 3.5
-    num_inference_steps: int = 30
+    num_inference_steps: int = 4
     seed: int = 0