PseudoTerminal X commited on
Commit
7791eb8
1 Parent(s): 5daaa9c

Trained for 1 epochs and 35500 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: 35000
66
  - Learning rate: 1e-06
67
  - Effective batch size: 192
68
  - Micro-batch size: 24
@@ -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: 4
228
  - Resolution: 1.0 megapixels
229
  - Cropped: False
230
  - 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: 4
244
  - Resolution: 1.0 megapixels
245
  - Cropped: False
246
  - Crop style: None
 
62
  ## Training settings
63
 
64
  - Training epochs: 1
65
+ - Training steps: 35500
66
  - Learning rate: 1e-06
67
  - Effective batch size: 192
68
  - Micro-batch size: 24
 
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: 2
228
  - Resolution: 1.0 megapixels
229
  - Cropped: False
230
  - Crop style: None
 
240
  ### nijijourney-v6-520k-raw
241
  - Repeats: 0
242
  - Total number of images: ~516288
243
+ - Total number of aspect buckets: 3
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:a0c2796621c2849512d6007b139950be8bad2cfada9ca43de8e3227f2d978162
3
  size 5451415117
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa1a7136c92558908a1c2b81c1c68bf828a8ca0984bd5c6ee745652a48f25eff
3
  size 5451415117
random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c23e7c622a74288766dea07bd66de1073b7fd7a88ba3bdf5d2159e097cf42aff
3
- size 16036
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:658413fe745cd97a7b1fd9d7676b34d9251f476c9326705bda6bfeba53ca5ebc
3
+ size 16100
scheduler.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b9a9723ad329887a64c1a37c3a7336347f4f420b46eb6660585dcca99720703c
3
  size 1000
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5cf3fffa2c56666db1a062e04ba11466ec4b818ef4244dc94590d18c86c95e55
3
  size 1000
training_state-anatomy.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:26c062612bb87480b5ecc18ea348478e1b24d4f6eb983b68460e85d8c90aed66
3
- size 10465958
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:99b6f5d28783c45361aa0cd9dd799ae63369803d932f075f9e321bdca3c4e83f
3
+ size 10606242
training_state-midjourney-v6-520k-raw.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3ad1e7eaaee901177073068979dc916a7d14792bc184afd8473118478fbf83cd
3
- size 8737551
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b24917d8eb7f09275a9135b43c2c9b4ed0e1e9c45baf9b9ff8d4b572d4af921c
3
+ size 8927895
training_state-nijijourney-v6-520k-raw.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:347fcf13111e6b75ad1e33d50d8f4bb02b1788137b2135622881941351e062a4
3
- size 9189091
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a3d2685e7aaf3a1bfbfa316205786cc9b40b9539d3ee7690a33be0cc4f554e7
3
+ size 9421411
training_state-photo-concept-bucket.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:24176cce40b1acdb6c4e26dca2843807dc487263c54e85acda212c9ace2145c3
3
- size 7075938
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:944d863ab1881c90e32bd86c614b2a504a05bbdefd5d88274b2213aed937442f
3
+ size 7240538
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": 35000, "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", "bg20k-1024"], "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": 13, "shutterstock": 0, "nsfw-1024": 0, "photo-aesthetics": 0, "anatomy": 5, "bg20k-1024": 0, "sfwbooru": 0, "midjourney-v6-520k-raw": 0, "nijijourney-v6-520k-raw": 0, "photo-concept-bucket": 0, "dalle3": 0}}
 
1
+ {"global_step": 35500, "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", "bg20k-1024", "anatomy"], "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": 14, "shutterstock": 0, "nsfw-1024": 0, "photo-aesthetics": 0, "anatomy": 0, "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:aef80d4100f990ad3a9d7360f9987bba3a554d5e33f4e9225fa01998f938d5d4
3
  size 1816969728
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:66e6cd23cc38bac6623be10f5e81913b235721416fc0ee35a283fed1f99aad44
3
  size 1816969728