davidrd123
commited on
Commit
•
a07a411
1
Parent(s):
c343954
Model card auto-generated by SimpleTuner
Browse files
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:
|
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=
|
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%
|