Upload task output f9972e57-22f3-4124-8a34-02230a811b7e

last/config.yaml

@@ -33,10 +33,19 @@ config:
       cache_text_embeddings: true
       dtype: bf16
       gradient_checkpointing: true
-      lr: 0.0001
+      lr: 1.0
+      model_name: gradients-io-tournaments/Qwen-Image
       noise_scheduler: flowmatch
-      optimizer: adamw8bit
-      steps: 3000
+      optimizer: prodigy
+      optimizer_params:
+        d_coef: 1.0
+        decouple: true
+        safeguard_warmup: true
+        use_bias_correction: true
+        weight_decay: 0.01
+      save_every: 325
+      save_every_n_epochs: 40
+      steps: 3600
       timestep_type: weighted
       train_text_encoder: false
       train_unet: true