davidrd123 commited on
Commit
a07a411
1 Parent(s): c343954

Model card auto-generated by SimpleTuner

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -94,7 +94,7 @@ You may reuse the base model text encoder for inference.
94
  ## Training settings
95
 
96
  - Training epochs: 17
97
- - Training steps: 7000
98
  - Learning rate: 0.0001
99
  - Learning rate schedule: polynomial
100
  - Warmup steps: 200
@@ -104,7 +104,7 @@ You may reuse the base model text encoder for inference.
104
  - Gradient accumulation steps: 1
105
  - Number of GPUs: 1
106
  - Gradient checkpointing: True
107
- - Prediction type: flow-matching (extra parameters=['flux_schedule_auto_shift', 'shift=0.0', 'flux_guidance_mode=constant', 'flux_guidance_value=1.0', 'flux_beta_schedule_alpha=5.0', 'flux_beta_schedule_beta=2.0', 'flow_matching_loss=compatible'])
108
  - Optimizer: adamw_bf16
109
  - Trainable parameter precision: Pure BF16
110
  - Caption dropout probability: 10.0%
 
94
  ## Training settings
95
 
96
  - Training epochs: 17
97
+ - Training steps: 7250
98
  - Learning rate: 0.0001
99
  - Learning rate schedule: polynomial
100
  - Warmup steps: 200
 
104
  - Gradient accumulation steps: 1
105
  - Number of GPUs: 1
106
  - Gradient checkpointing: True
107
+ - Prediction type: flow-matching (extra parameters=['flux_schedule_auto_shift', 'shift=0.0', 'flux_guidance_mode=constant', 'flux_guidance_value=1.0', 'flux_beta_schedule_alpha=2.0', 'flux_beta_schedule_beta=8.0', 'flow_matching_loss=compatible'])
108
  - Optimizer: adamw_bf16
109
  - Trainable parameter precision: Pure BF16
110
  - Caption dropout probability: 10.0%