PseudoTerminal X commited on
Commit
25838a3
·
verified ·
1 Parent(s): 2333850

Trained for 5 epochs and 201000 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: 200000
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: 201000
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:00f11c59fa7170487391f57203d6b631f07c16123724b60845ffd658f5d74c09
3
  size 5451415117
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d175e466ccae4e326dd1b40dff3c62b2d4957835b2533769e371e4f177861f52
3
  size 5451415117
random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:81dfe2a708c55c46075ad4f141c8e9f9094c3533f4d033e055857e6755d9fb4e
3
- size 16164
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c8b01ae362aa2c1fcd4eb556c8655fcacfa88e61ce00587094db0a38b0b9617a
3
+ size 16036
scheduler.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:29a6b23f4e539806186c4b2e2c941262b513fcd7e55f94649966cd795e4309de
3
  size 1000
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4fd6469ada013bf476b5e046e03e8c5fe02e1e04da19db8e59fcb2f65bb6a72a
3
  size 1000
training_state-midjourney-v6-520k-raw.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:080949af93b2910d1e7e3cb773c52ef82bc6ba2010c6738b945300307e54ca28
3
- size 4633810
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df9c411c8d1fabc67d307f51366c664fd0256a928887b6d5027983e41f023af5
3
+ size 4787722
training_state-nijijourney-v6-520k-raw.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:243c1725364e02e8b336ef70ce3210043998b9e45d3e6bf10319e8f46b344a80
3
- size 4789682
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:add28c1571e7049b47fc34b674f6b1962dd147ab55cb6cfaa7569d01030f83cf
3
+ size 4926210
training_state-sfwbooru.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state.json CHANGED
@@ -1 +1 @@
1
- {"global_step": 200000, "epoch_step": 200005, "epoch": 6, "exhausted_backends": ["signs", "guys", "nijijourney", "bookcovers", "celebrities", "normalnudes", "sports", "moviecollection", "gay", "experimental", "yoga", "cinemamix-1mp", "ethnic", "architecture", "shutterstock", "nsfw-1024", "photo-aesthetics", "id-75k", "text-1mp", "bg20k-1024", "mj-60", "anatomy", "dalle3"], "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": 0, "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": 201000, "epoch_step": 201005, "epoch": 6, "exhausted_backends": ["signs", "guys", "nijijourney", "bookcovers", "celebrities", "normalnudes", "sports", "moviecollection", "gay", "experimental", "yoga", "cinemamix-1mp", "ethnic", "architecture", "shutterstock", "nsfw-1024", "photo-aesthetics", "id-75k", "text-1mp", "bg20k-1024", "mj-60", "anatomy", "dalle3"], "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": 0, "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:de24d8549b23a0ba5f752bc4efb263412a1e407c1ad42487baf4dbf13f0d258c
3
  size 1816969728
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d7b8f92933c1ea9a4fcfe861fc99062aba783ec398b9c5d7057966bc4d2f104d
3
  size 1816969728