Model card auto-generated by SimpleTuner

README.md

@@ -87,13 +87,13 @@ You may reuse the base model text encoder for inference.
 ## Training settings
 
 - Training epochs: 9
-- Training steps: 110
+- Training steps: 100
 - Learning rate: 1.0
 - Effective batch size: 1
   - Micro-batch size: 1
   - Gradient accumulation steps: 1
   - Number of GPUs: 1
-- Prediction type: epsilon
+- Prediction type: flow-matching
 - Rescaled betas zero SNR: False
 - Optimizer: Prodigy
 - Precision: no
@@ -136,7 +136,7 @@ image = pipeline(
     prompt=prompt,
     num_inference_steps=28,
     generator=torch.Generator(device='cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu').manual_seed(1641421826),
-        width=1024,
+    width=1024,
     height=1024,
     guidance_scale=3.0,
 ).images[0]