PseudoTerminal X commited on
Commit
23e7239
·
verified ·
1 Parent(s): 6489c33

Trained for 5 epochs and 174000 steps.

Browse files

Trained with datasets ['text-embeds', 'text-embeds-extra', 'image-embeds', 'sports', 'mj-60', 'id-75k', 'celebrities', 'normalnudes', 'guys', 'signs', 'dalle3', 'sfwbooru', 'moviecollection', 'bookcovers', 'nijijourney', 'experimental', 'ethnic', 'gay', 'architecture', 'shutterstock', 'midjourney-v6-520k-raw', 'nijijourney-v6-520k-raw', 'cinemamix-1mp', 'nsfw-1024', 'anatomy', 'bg20k-1024', 'yoga', 'photo-aesthetics', 'text-1mp']
Learning rate 1e-06, batch size 8, 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-v0.7-stage1
VAE: ptx0/pixart-900m-1024-ft-v0.7-stage1

README.md CHANGED
@@ -552,7 +552,7 @@ You may reuse the base model text encoder for inference.
552
  ## Training settings
553
 
554
  - Training epochs: 5
555
- - Training steps: 173000
556
  - Learning rate: 1e-06
557
  - Effective batch size: 64
558
  - Micro-batch size: 8
 
552
  ## Training settings
553
 
554
  - Training epochs: 5
555
+ - Training steps: 174000
556
  - Learning rate: 1e-06
557
  - Effective batch size: 64
558
  - Micro-batch size: 8
optimizer.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c95fb0362f016c64e469dbddfe2acbd424d8bbd23603f9453ece7bc03acb2600
3
  size 5451415117
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bcf1e27aaa0b64f3882a180222c21be19f0e617dd594421ce80b7863cbbeea22
3
  size 5451415117
random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:306547e63d6907acc6c96c917d8e16d925e5a503f235dc89e4bc42ad8ebd9237
3
- size 16100
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8e97c5637e7976cc43f95fccfda552ed0f4730f32a049273225f110f2fe4d5f5
3
+ size 16036
scheduler.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f985f173b347a7c026ade9c3c5720a5b289f2a5a7e255415b40c78519a088785
3
  size 1000
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:71be2f1014f37c7a42fd5b08a68e86ab17f99b72de0bd26b474c5dd4673f931d
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:f76aae3d44a5336fc659c79bac70f7d6d3a009e625c9da43ce705e44b1f87e1c
3
- size 14954825
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4fd1f18d0273aef8016d04c069be53fbf693974f0971328f76137612c826bc2c
3
+ size 15002118
training_state-id-75k.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-midjourney-v6-520k-raw.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2942ac511f54e6b4dd3d3ae281a0c9ddd7668fc4b5c6c47e6a49070a35db3df7
3
- size 2395090
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2dfec7c3f051ac6a12e271ac531bbcffc2e7cb527aeb95845a1d05718f4e8434
3
+ size 2437490
training_state-mj-60.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-nijijourney-v6-520k-raw.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:04d1e29be5c6fd6f70ffb546feed1102b0a80c7812341e14b5bc49c68db770d9
3
- size 2550114
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f497769fe431a8f19416b7250da63dfb307f8313fc291d6f27a2ad0a5db7ed6c
3
+ size 2587426
training_state-nsfw-1024.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-photo-aesthetics.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-sfwbooru.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-shutterstock.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": 173000, "epoch_step": 173005, "epoch": 6, "exhausted_backends": ["signs", "guys", "nijijourney", "bookcovers", "celebrities", "normalnudes", "sports", "moviecollection", "gay", "experimental", "yoga", "cinemamix-1mp", "ethnic", "architecture"], "repeats": {"nijijourney": 0, "signs": 0, "guys": 0, "celebrities": 0, "bookcovers": 0, "normalnudes": 0, "moviecollection": 0, "sports": 0, "gay": 0, "experimental": 0, "yoga": 0, "ethnic": 0, "architecture": 0, "cinemamix-1mp": 0, "photo-aesthetics": 0, "anatomy": 0, "text-1mp": 1, "nsfw-1024": 0, "shutterstock": 0, "id-75k": 0, "bg20k-1024": 0, "mj-60": 0, "sfwbooru": 0, "midjourney-v6-520k-raw": 0, "nijijourney-v6-520k-raw": 0, "dalle3": 0}}
 
1
+ {"global_step": 174000, "epoch_step": 174005, "epoch": 6, "exhausted_backends": ["signs", "guys", "nijijourney", "bookcovers", "celebrities", "normalnudes", "sports", "moviecollection", "gay", "experimental", "yoga", "cinemamix-1mp", "ethnic", "architecture"], "repeats": {"nijijourney": 0, "signs": 0, "guys": 0, "celebrities": 0, "bookcovers": 0, "normalnudes": 0, "moviecollection": 0, "sports": 0, "gay": 0, "experimental": 0, "yoga": 0, "ethnic": 0, "architecture": 0, "cinemamix-1mp": 0, "photo-aesthetics": 0, "anatomy": 0, "text-1mp": 1, "nsfw-1024": 0, "shutterstock": 0, "id-75k": 0, "bg20k-1024": 0, "mj-60": 0, "sfwbooru": 0, "midjourney-v6-520k-raw": 0, "nijijourney-v6-520k-raw": 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:ea1151aa90b82cff9184075ee6e39e3d413a82af477fd1cc7f7d5bd63c7e5c72
3
  size 1816969728
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:26dae8a4c9a8cbba0b1c9ca2476a41865f4b40e60aeece03987c831e500b7682
3
  size 1816969728