davidrd123 commited on
Commit
6da798e
·
verified ·
1 Parent(s): e747ae5

Model card auto-generated by SimpleTuner

Browse files
Files changed (1) hide show
  1. README.md +8 -8
README.md CHANGED
@@ -77,7 +77,7 @@ None
77
  - Steps: `20`
78
  - Sampler: `FlowMatchEulerDiscreteScheduler`
79
  - Seed: `42`
80
- - Resolution: `1024x1024`
81
  - Skip-layer guidance:
82
 
83
  Note: The validation settings are not necessarily the same as the [training settings](#training-settings).
@@ -93,10 +93,10 @@ You may reuse the base model text encoder for inference.
93
 
94
  ## Training settings
95
 
96
- - Training epochs: 8
97
- - Training steps: 5000
98
- - Learning rate: 0.0003
99
- - Learning rate schedule: polynomial
100
  - Warmup steps: 200
101
  - Max grad norm: 2.0
102
  - Effective batch size: 3
@@ -167,7 +167,7 @@ You may reuse the base model text encoder for inference.
167
  ### frankpape-1536
168
  - Repeats: 5
169
  - Total number of images: 16
170
- - Total number of aspect buckets: 2
171
  - Resolution: 2.359296 megapixels
172
  - Cropped: False
173
  - Crop style: None
@@ -249,8 +249,8 @@ image = pipeline(
249
  prompt=prompt,
250
  num_inference_steps=20,
251
  generator=torch.Generator(device='cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu').manual_seed(42),
252
- width=1024,
253
- height=1024,
254
  guidance_scale=3.0,
255
  ).images[0]
256
  image.save("output.png", format="PNG")
 
77
  - Steps: `20`
78
  - Sampler: `FlowMatchEulerDiscreteScheduler`
79
  - Seed: `42`
80
+ - Resolution: `896x1312`
81
  - Skip-layer guidance:
82
 
83
  Note: The validation settings are not necessarily the same as the [training settings](#training-settings).
 
93
 
94
  ## Training settings
95
 
96
+ - Training epochs: 9
97
+ - Training steps: 5100
98
+ - Learning rate: 0.0001
99
+ - Learning rate schedule: constant
100
  - Warmup steps: 200
101
  - Max grad norm: 2.0
102
  - Effective batch size: 3
 
167
  ### frankpape-1536
168
  - Repeats: 5
169
  - Total number of images: 16
170
+ - Total number of aspect buckets: 1
171
  - Resolution: 2.359296 megapixels
172
  - Cropped: False
173
  - Crop style: None
 
249
  prompt=prompt,
250
  num_inference_steps=20,
251
  generator=torch.Generator(device='cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu').manual_seed(42),
252
+ width=896,
253
+ height=1312,
254
  guidance_scale=3.0,
255
  ).images[0]
256
  image.save("output.png", format="PNG")