davidrd123 commited on
Commit
9b49fbf
1 Parent(s): ac2f450

Model card auto-generated by SimpleTuner

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -94,8 +94,8 @@ You may reuse the base model text encoder for inference.
94
  ## Training settings
95
 
96
  - Training epochs: 11
97
- - Training steps: 4500
98
- - Learning rate: 0.0003
99
  - Learning rate schedule: polynomial
100
  - Warmup steps: 200
101
  - Max grad norm: 2.0
@@ -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=2.0', 'flux_beta_schedule_beta=5.0', 'flow_matching_loss=compatible'])
108
  - Optimizer: adamw_bf16
109
  - Trainable parameter precision: Pure BF16
110
  - Caption dropout probability: 10.0%
@@ -140,7 +140,7 @@ You may reuse the base model text encoder for inference.
140
  ### cassatt-512
141
  - Repeats: 10
142
  - Total number of images: 16
143
- - Total number of aspect buckets: 6
144
  - Resolution: 0.262144 megapixels
145
  - Cropped: False
146
  - Crop style: None
 
94
  ## Training settings
95
 
96
  - Training epochs: 11
97
+ - Training steps: 4750
98
+ - Learning rate: 0.0001
99
  - Learning rate schedule: polynomial
100
  - Warmup steps: 200
101
  - Max grad norm: 2.0
 
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%
 
140
  ### cassatt-512
141
  - Repeats: 10
142
  - Total number of images: 16
143
+ - Total number of aspect buckets: 1
144
  - Resolution: 0.262144 megapixels
145
  - Cropped: False
146
  - Crop style: None