davidrd123
commited on
Commit
•
d5074e2
1
Parent(s):
085eaee
Model card auto-generated by SimpleTuner
Browse files
README.md
CHANGED
@@ -94,15 +94,15 @@ You may reuse the base model text encoder for inference.
|
|
94 |
|
95 |
## Training settings
|
96 |
|
97 |
-
- Training epochs:
|
98 |
-
- Training steps:
|
99 |
- Learning rate: 7e-05
|
100 |
- Max grad norm: 0.1
|
101 |
- Effective batch size: 3
|
102 |
- Micro-batch size: 3
|
103 |
- Gradient accumulation steps: 1
|
104 |
- Number of GPUs: 1
|
105 |
-
- Prediction type: flow-matching (extra parameters=['shift=
|
106 |
- Rescaled betas zero SNR: False
|
107 |
- Optimizer: adamw_bf16
|
108 |
- Precision: Pure BF16
|
|
|
94 |
|
95 |
## Training settings
|
96 |
|
97 |
+
- Training epochs: 2
|
98 |
+
- Training steps: 1500
|
99 |
- Learning rate: 7e-05
|
100 |
- Max grad norm: 0.1
|
101 |
- Effective batch size: 3
|
102 |
- Micro-batch size: 3
|
103 |
- Gradient accumulation steps: 1
|
104 |
- Number of GPUs: 1
|
105 |
+
- Prediction type: flow-matching (extra parameters=['shift=2.0', 'flux_guidance_value=1.0'])
|
106 |
- Rescaled betas zero SNR: False
|
107 |
- Optimizer: adamw_bf16
|
108 |
- Precision: Pure BF16
|