PseudoTerminal X commited on
Commit
15651bc
·
verified ·
1 Parent(s): 62e421c

Trained for 1 epochs and 31000 steps.

Browse files

Trained with datasets ['text-embeds-pixart-filter', 'photo-concept-bucket', 'moviecollection', 'experimental', 'ethnic', 'sports', 'architecture', 'shutterstock', 'cinemamix-1mp', 'nsfw-1024', 'anatomy', 'bg20k-1024', 'yoga', 'photo-aesthetics', 'text-1mp', 'movieposters', 'normalnudes', 'pixel-art', 'signs', 'midjourney-v6-520k-raw', 'sfwbooru', 'nijijourney-v6-520k-raw', 'dalle3']
Learning rate 1e-06, batch size 24, and 1 gradient accumulation steps.
Used DDPM noise scheduler for training with epsilon prediction type and rescaled_betas_zero_snr=False
Using 'linspace' timestep spacing.
Base model: ptx0/pixart-900m-1024-ft-large
VAE: madebyollin/sdxl-vae-fp16-fix

README.md CHANGED
@@ -62,7 +62,7 @@ You may reuse the base model text encoder for inference.
62
  ## Training settings
63
 
64
  - Training epochs: 1
65
- - Training steps: 30500
66
  - Learning rate: 1e-06
67
  - Effective batch size: 192
68
  - Micro-batch size: 24
@@ -80,7 +80,7 @@ You may reuse the base model text encoder for inference.
80
  ### photo-concept-bucket
81
  - Repeats: 0
82
  - Total number of images: ~564672
83
- - Total number of aspect buckets: 8
84
  - Resolution: 1.0 megapixels
85
  - Cropped: False
86
  - Crop style: None
@@ -152,7 +152,7 @@ You may reuse the base model text encoder for inference.
152
  ### anatomy
153
  - Repeats: 5
154
  - Total number of images: ~15168
155
- - Total number of aspect buckets: 2
156
  - Resolution: 1.0 megapixels
157
  - Cropped: True
158
  - Crop style: random
@@ -224,7 +224,7 @@ You may reuse the base model text encoder for inference.
224
  ### midjourney-v6-520k-raw
225
  - Repeats: 0
226
  - Total number of images: ~513792
227
- - Total number of aspect buckets: 10
228
  - Resolution: 1.0 megapixels
229
  - Cropped: False
230
  - Crop style: None
@@ -232,7 +232,7 @@ You may reuse the base model text encoder for inference.
232
  ### sfwbooru
233
  - Repeats: 0
234
  - Total number of images: ~271488
235
- - Total number of aspect buckets: 22
236
  - Resolution: 1.0 megapixels
237
  - Cropped: False
238
  - Crop style: None
@@ -240,7 +240,7 @@ You may reuse the base model text encoder for inference.
240
  ### nijijourney-v6-520k-raw
241
  - Repeats: 0
242
  - Total number of images: ~516288
243
- - Total number of aspect buckets: 8
244
  - Resolution: 1.0 megapixels
245
  - Cropped: False
246
  - Crop style: None
 
62
  ## Training settings
63
 
64
  - Training epochs: 1
65
+ - Training steps: 31000
66
  - Learning rate: 1e-06
67
  - Effective batch size: 192
68
  - Micro-batch size: 24
 
80
  ### photo-concept-bucket
81
  - Repeats: 0
82
  - Total number of images: ~564672
83
+ - Total number of aspect buckets: 7
84
  - Resolution: 1.0 megapixels
85
  - Cropped: False
86
  - Crop style: None
 
152
  ### anatomy
153
  - Repeats: 5
154
  - Total number of images: ~15168
155
+ - Total number of aspect buckets: 3
156
  - Resolution: 1.0 megapixels
157
  - Cropped: True
158
  - Crop style: random
 
224
  ### midjourney-v6-520k-raw
225
  - Repeats: 0
226
  - Total number of images: ~513792
227
+ - Total number of aspect buckets: 9
228
  - Resolution: 1.0 megapixels
229
  - Cropped: False
230
  - Crop style: None
 
232
  ### sfwbooru
233
  - Repeats: 0
234
  - Total number of images: ~271488
235
+ - Total number of aspect buckets: 20
236
  - Resolution: 1.0 megapixels
237
  - Cropped: False
238
  - Crop style: None
 
240
  ### nijijourney-v6-520k-raw
241
  - Repeats: 0
242
  - Total number of images: ~516288
243
+ - Total number of aspect buckets: 7
244
  - Resolution: 1.0 megapixels
245
  - Cropped: False
246
  - Crop style: None
optimizer.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6948725c906d45237d9d2c788459cb66dfbcbf3e0067aa454abae2e1dbce4afe
3
  size 5451415117
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03b9be4c2a3c8636ca491e90254b28ce779d9a47672f79f12103acd781627912
3
  size 5451415117
random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7f41d4909c8584fb56528c2c33a7bec17370b9848715c4fd74d50e20e274a4d6
3
- size 16100
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b84649c102e44f1d75bea64e89138712bad0e9795dc1664ea2867358c03c8e78
3
+ size 16036
scheduler.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4ebe1252aa3efaeb89cd726d000ab16e13f8dd6867f5505fde197fd2546513a6
3
  size 1000
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:848fca831655e283b7d6b1850a72f4357950642db44279410f174a1c56b2a903
3
  size 1000
training_state-anatomy.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-bg20k-1024.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-dalle3.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:dd6a18698942d4d7fc10272e2f2b77e5633cd2235cbaa2be1c405206931993e5
3
- size 9346773
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bbb0d564c0f2a8aa3b256c7365f4de38ff2f5ea1fdb4276027f249cabff946de
3
+ size 9452108
training_state-midjourney-v6-520k-raw.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:05ddf85971f9d05e6c473f3a341386f6f2907e7e7afa7de38672689a19f7811c
3
- size 6950295
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e4b38053fe0a6ee9f6ec99bb71edf9e6d3f4e1de54c5568696dbe79ca7f00b61
3
+ size 7130751
training_state-nijijourney-v6-520k-raw.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e1926de0877ddf18bf61d6747de63354708ed4418d6888af0ef12c1c8bb33741
3
- size 7438771
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5166f281282bcdecf5b893fb47f23d5884d670c16ba7f01cc0e94c90711c62b8
3
+ size 7636771
training_state-photo-concept-bucket.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3fb388d9a6037c78b10fa49b47ccfc5dc6e660b62fcfe773243a3c3c9f0010e8
3
- size 5783802
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c32bf35104a4368c07723e4d899932e7caf9067749bb72f68d7851887a04050e
3
+ size 5911706
training_state-sfwbooru.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-text-1mp.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state.json CHANGED
@@ -1 +1 @@
1
- {"global_step": 30500, "epoch_step": 1, "epoch": 2, "exhausted_backends": ["pixel-art", "signs", "sports", "ethnic", "experimental", "movieposters", "normalnudes", "yoga", "cinemamix-1mp", "architecture", "moviecollection", "shutterstock", "nsfw-1024", "photo-aesthetics"], "repeats": {"bookcovers": 0, "signs": 0, "normalnudes": 0, "nijijourney": 0, "movieposters": 0, "celebrities": 0, "pixel-art": 0, "propagandaposters": 0, "sports": 0, "moviecollection": 0, "gay": 0, "experimental": 0, "yoga": 0, "ethnic": 0, "cinemamix-1mp": 0, "architecture": 0, "mj-60": 0, "text-1mp": 4, "shutterstock": 0, "nsfw-1024": 0, "photo-aesthetics": 0, "anatomy": 1, "bg20k-1024": 0, "sfwbooru": 0, "midjourney-v6-520k-raw": 0, "nijijourney-v6-520k-raw": 0, "photo-concept-bucket": 0, "dalle3": 0}}
 
1
+ {"global_step": 31000, "epoch_step": 1, "epoch": 2, "exhausted_backends": ["pixel-art", "signs", "sports", "ethnic", "experimental", "movieposters", "normalnudes", "yoga", "cinemamix-1mp", "architecture", "moviecollection", "shutterstock", "nsfw-1024", "photo-aesthetics"], "repeats": {"bookcovers": 0, "signs": 0, "normalnudes": 0, "nijijourney": 0, "movieposters": 0, "celebrities": 0, "pixel-art": 0, "propagandaposters": 0, "sports": 0, "moviecollection": 0, "gay": 0, "experimental": 0, "yoga": 0, "ethnic": 0, "cinemamix-1mp": 0, "architecture": 0, "mj-60": 0, "text-1mp": 5, "shutterstock": 0, "nsfw-1024": 0, "photo-aesthetics": 0, "anatomy": 2, "bg20k-1024": 0, "sfwbooru": 0, "midjourney-v6-520k-raw": 0, "nijijourney-v6-520k-raw": 0, "photo-concept-bucket": 0, "dalle3": 0}}
transformer/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5d4d97f4bbdf3d34adfb319d94fc7999bfa5d97cd2c578d44a343829eaca3c8a
3
  size 1816969728
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:136147de23e23bfd8aa9fd455c9229d25c76dbdc7389447fac9ee24c44e9fc37
3
  size 1816969728