JiaxiJiang commited on
Commit
aa82a9a
·
verified ·
1 Parent(s): 8fea66c

End of training

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. README.md +39 -0
  2. checkpoint-1000/model.safetensors +3 -0
  3. checkpoint-1000/optimizer.bin +3 -0
  4. checkpoint-1000/random_states_0.pkl +3 -0
  5. checkpoint-1000/scheduler.bin +3 -0
  6. checkpoint-1500/model.safetensors +3 -0
  7. checkpoint-1500/optimizer.bin +3 -0
  8. checkpoint-1500/random_states_0.pkl +3 -0
  9. checkpoint-1500/scheduler.bin +3 -0
  10. checkpoint-2000/model.safetensors +3 -0
  11. checkpoint-2000/optimizer.bin +3 -0
  12. checkpoint-2000/random_states_0.pkl +3 -0
  13. checkpoint-2000/scheduler.bin +3 -0
  14. checkpoint-2500/model.safetensors +3 -0
  15. checkpoint-2500/optimizer.bin +3 -0
  16. checkpoint-2500/random_states_0.pkl +3 -0
  17. checkpoint-2500/scheduler.bin +3 -0
  18. checkpoint-3000/model.safetensors +3 -0
  19. checkpoint-3000/optimizer.bin +3 -0
  20. checkpoint-3000/random_states_0.pkl +3 -0
  21. checkpoint-3000/scheduler.bin +3 -0
  22. checkpoint-3500/model.safetensors +3 -0
  23. checkpoint-3500/optimizer.bin +3 -0
  24. checkpoint-3500/random_states_0.pkl +3 -0
  25. checkpoint-3500/scheduler.bin +3 -0
  26. checkpoint-4000/model.safetensors +3 -0
  27. checkpoint-4000/optimizer.bin +3 -0
  28. checkpoint-4000/random_states_0.pkl +3 -0
  29. checkpoint-4000/scheduler.bin +3 -0
  30. checkpoint-500/model.safetensors +3 -0
  31. checkpoint-500/optimizer.bin +3 -0
  32. checkpoint-500/random_states_0.pkl +3 -0
  33. checkpoint-500/scheduler.bin +3 -0
  34. feature_extractor/preprocessor_config.json +28 -0
  35. learned_embeds-steps-1000.safetensors +3 -0
  36. learned_embeds-steps-1500.safetensors +3 -0
  37. learned_embeds-steps-2000.safetensors +3 -0
  38. learned_embeds-steps-2500.safetensors +3 -0
  39. learned_embeds-steps-3000.safetensors +3 -0
  40. learned_embeds-steps-3500.safetensors +3 -0
  41. learned_embeds-steps-4000.safetensors +3 -0
  42. learned_embeds-steps-500.safetensors +3 -0
  43. learned_embeds.safetensors +3 -0
  44. logs/textual_inversion/1710426474.991172/events.out.tfevents.1710426474.slabgpu07.3826029.1 +3 -0
  45. logs/textual_inversion/1710426475.006273/hparams.yml +48 -0
  46. logs/textual_inversion/events.out.tfevents.1710426474.slabgpu07.3826029.0 +3 -0
  47. model_index.json +38 -0
  48. safety_checker/config.json +168 -0
  49. safety_checker/model.safetensors +3 -0
  50. scheduler/scheduler_config.json +15 -0
README.md ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: creativeml-openrail-m
3
+ library_name: diffusers
4
+ tags:
5
+ - stable-diffusion
6
+ - stable-diffusion-diffusers
7
+ - text-to-image
8
+ - diffusers
9
+ - textual_inversion
10
+ - diffusers-training
11
+ base_model: runwayml/stable-diffusion-v1-5
12
+ inference: true
13
+ ---
14
+
15
+ <!-- This model card has been generated automatically according to the information the training script had access to. You
16
+ should probably proofread and complete it, then remove this comment. -->
17
+
18
+
19
+ # Textual inversion text2image fine-tuning - JiaxiJiang/textual_inversion_duck2
20
+ These are textual inversion adaption weights for runwayml/stable-diffusion-v1-5. You can find some example images in the following.
21
+
22
+
23
+
24
+
25
+ ## Intended uses & limitations
26
+
27
+ #### How to use
28
+
29
+ ```python
30
+ # TODO: add an example code snippet for running this diffusion pipeline
31
+ ```
32
+
33
+ #### Limitations and bias
34
+
35
+ [TODO: provide examples of latent issues and potential remediations]
36
+
37
+ ## Training details
38
+
39
+ [TODO: describe the data used to train the model]
checkpoint-1000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:00d5d3d5c159a499c8938414ca49e2f196edf0b77c435c1ef1fc6e15a946c3d8
3
+ size 492272024
checkpoint-1000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:72de479577aab052b1ed8771840f4999be4d0284c9b84f751386c49239aae52b
3
+ size 303577058
checkpoint-1000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:56d682444aa67b9dcca70c54f5ba7f0727376d4614980e888abd7d7154ec6e17
3
+ size 15060
checkpoint-1000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3f0d829441992ff2ab1f7502a3a94afc011706461052e42e18e5b963678527c4
3
+ size 1000
checkpoint-1500/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:06107c0d6d70ca49f23ea3f321c35903facb68a5146837412f6156ca15957962
3
+ size 492272024
checkpoint-1500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:145e4800df70c98125bacf1d88193102774ee497af0b77e615a16b42abd3d029
3
+ size 303577058
checkpoint-1500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88cb0e1cf10375274e3ba707d95701e4ea5fa96bd9f0b53c4f9c599e138d138c
3
+ size 15060
checkpoint-1500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0746407d5e6c8d929c4e940db747c88272550cab75cfb856468297d71d7802ca
3
+ size 1000
checkpoint-2000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:04bc252372e827fcc1863404e0dee42c9be78cac78366808dbaf76acd235606c
3
+ size 492272024
checkpoint-2000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:77482eaaa9ffa20f0b163e1d5b90143634770ef86ea996b0db24524bec8f5044
3
+ size 303577058
checkpoint-2000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:75ea04ccc7465f4f5f4e5269917fa14e8c58fc2db673634b6a9e5a335756cfa9
3
+ size 15060
checkpoint-2000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5484c6ff4bca55d86f0d54575371a1f3b26246b4ff863161800c9be8539d8c77
3
+ size 1000
checkpoint-2500/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:17ac8765a6587fa1cacb30656b896ee466ca0ce841501d8b20eec05bb736bec3
3
+ size 492272024
checkpoint-2500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:655efe3856d5cc1391ad00e0e5723523444872956275d264285412b0d5cf3d2e
3
+ size 303577058
checkpoint-2500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:79e9ac22dd91d66e8c67058a45beb329f18b8b90b9952038ac45268d7c89c176
3
+ size 15060
checkpoint-2500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0bb1a67b9696a2697693774a6f09d8186c78e37cff7896931a3632a6822f70b9
3
+ size 1000
checkpoint-3000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:653a4cebfcd45709aeabd990c4a79a8db8b4d0daf620b39b1adb603bcc869a24
3
+ size 492272024
checkpoint-3000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4f3e8118393a4c36d8573443e8ae8c0bf691914ea26ae631dc637bf34bebb248
3
+ size 303577058
checkpoint-3000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88cd1e1b5e3c65eff4cbea9e50e124a2b11d8264701cc114ffcebefea5ccad65
3
+ size 15124
checkpoint-3000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2ed1284c2bacec58dd33ce8bc7070e7691d6d084b54330b85bacc37784d59cee
3
+ size 1000
checkpoint-3500/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c3b48a7641984ae2c7b80ce29b0bab1b282e3fefe7eb26be975ec08db81a1699
3
+ size 492272024
checkpoint-3500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2ac9f6762a3f8139d11f11f9272a6876d5447392bfe06700854b156d17d686c4
3
+ size 303577058
checkpoint-3500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c937393fe12c5c9cb879f36bb6bf9eae1620609d118389a9b457e98fcdd4189f
3
+ size 15060
checkpoint-3500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7a7612087c5ad00e2b45e9904946f2338013b56bf6bfb0f7f47c6923d133581b
3
+ size 1000
checkpoint-4000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:05399cde303a580eab611ace3db61e86009ec8e3db9b415e128eeaa502fc499e
3
+ size 492272024
checkpoint-4000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2df6ebababd90f61b66d93e0830c3664a2896947503ea1f0408ad84f06563b3d
3
+ size 303577058
checkpoint-4000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dca4e7652f8daf768eff1a7f6b15ddd3a7e8466361c3b141a4df05108929c860
3
+ size 15060
checkpoint-4000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8653c1f480af4758444c6e3794dc046c1f7fb265db1c0d955846a2e97956b8e3
3
+ size 1000
checkpoint-500/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:45669caa14ad241c64221bc29b095feae5e0b41bdd7fed286fa7fa18a29c6f29
3
+ size 492272024
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6de418c0acddb09f0297ca0b1d432fb7ba4e19aaa21d7b90473fc1e1e565e9fd
3
+ size 303577058
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:92ed661703991e5ae576e48ec83e31e9df95a9e489806465cb72d9e328f8f270
3
+ size 15124
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8cad514bb36617120f19b7cde3b36a22bc8becf6cc6cfb2166811645c215cccb
3
+ size 1000
feature_extractor/preprocessor_config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "crop_size": {
3
+ "height": 224,
4
+ "width": 224
5
+ },
6
+ "do_center_crop": true,
7
+ "do_convert_rgb": true,
8
+ "do_normalize": true,
9
+ "do_rescale": true,
10
+ "do_resize": true,
11
+ "feature_extractor_type": "CLIPFeatureExtractor",
12
+ "image_mean": [
13
+ 0.48145466,
14
+ 0.4578275,
15
+ 0.40821073
16
+ ],
17
+ "image_processor_type": "CLIPImageProcessor",
18
+ "image_std": [
19
+ 0.26862954,
20
+ 0.26130258,
21
+ 0.27577711
22
+ ],
23
+ "resample": 3,
24
+ "rescale_factor": 0.00392156862745098,
25
+ "size": {
26
+ "shortest_edge": 224
27
+ }
28
+ }
learned_embeds-steps-1000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ec3d5f364a2f829260cf6a559b1c652518914f1041da28a87b2de7b0efce1046
3
+ size 6256
learned_embeds-steps-1500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61bf8a5bd8d5b30da181ab7caf15acc0867ea7d061b56f061aab47bd52d5f58b
3
+ size 6256
learned_embeds-steps-2000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a373aea2c654ad5fbacfe14b51a45894049ec4f7bfe7ee0c685730593bcfc303
3
+ size 6256
learned_embeds-steps-2500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f3915d44856bc70d2de7b4650b40babce2d95c4adeb0bc97a624d6f85210578
3
+ size 6256
learned_embeds-steps-3000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d0f978462a26d36b91758b72ac58be66cbaf795bbe010970022f89ee507f4d5f
3
+ size 6256
learned_embeds-steps-3500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb624a41bc000039e5b088df05ade70cb9ae34875061ecaa717361e2c0a02973
3
+ size 6256
learned_embeds-steps-4000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5522f7c7623b552ab461c0b9a06719b61119bcef8c40d55ec3e296ef61312a71
3
+ size 6256
learned_embeds-steps-500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f5c7fd8ac915d3d90c927d546a9fa954b4c1a5ab41b0fc8a8db8f048c8936c0
3
+ size 6256
learned_embeds.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5522f7c7623b552ab461c0b9a06719b61119bcef8c40d55ec3e296ef61312a71
3
+ size 6256
logs/textual_inversion/1710426474.991172/events.out.tfevents.1710426474.slabgpu07.3826029.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e7bf84b2307e0149c0b0ebdfb153f44e22184e02c27974d2bf4b338ebefe1014
3
+ size 2351
logs/textual_inversion/1710426475.006273/hparams.yml ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.01
5
+ allow_tf32: false
6
+ center_crop: false
7
+ checkpointing_steps: 500
8
+ checkpoints_total_limit: null
9
+ dataloader_num_workers: 0
10
+ enable_xformers_memory_efficient_attention: false
11
+ gradient_accumulation_steps: 4
12
+ gradient_checkpointing: false
13
+ hub_model_id: null
14
+ hub_token: null
15
+ initializer_token: duck
16
+ learnable_property: object
17
+ learning_rate: 0.002
18
+ local_rank: -1
19
+ logging_dir: logs
20
+ lr_num_cycles: 1
21
+ lr_scheduler: constant
22
+ lr_warmup_steps: 0
23
+ max_train_steps: 4000
24
+ mixed_precision: 'no'
25
+ no_safe_serialization: false
26
+ num_train_epochs: 40
27
+ num_validation_images: 4
28
+ num_vectors: 2
29
+ output_dir: textual_inversion_duck2
30
+ placeholder_token: <duck-toy>
31
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
32
+ push_to_hub: true
33
+ repeats: 100
34
+ report_to: tensorboard
35
+ resolution: 512
36
+ resume_from_checkpoint: null
37
+ revision: null
38
+ save_as_full_pipeline: false
39
+ save_steps: 500
40
+ scale_lr: true
41
+ seed: null
42
+ tokenizer_name: null
43
+ train_batch_size: 1
44
+ train_data_dir: ./dreambooth_dataset/duck_toy
45
+ validation_epochs: null
46
+ validation_prompt: null
47
+ validation_steps: 100
48
+ variant: null
logs/textual_inversion/events.out.tfevents.1710426474.slabgpu07.3826029.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:928adecb7a48166dd7a89649bb2d33bacfdd455c4bd434a2246c05734ed7b4b1
3
+ size 1343054
model_index.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "StableDiffusionPipeline",
3
+ "_diffusers_version": "0.27.0.dev0",
4
+ "_name_or_path": "runwayml/stable-diffusion-v1-5",
5
+ "feature_extractor": [
6
+ "transformers",
7
+ "CLIPImageProcessor"
8
+ ],
9
+ "image_encoder": [
10
+ null,
11
+ null
12
+ ],
13
+ "requires_safety_checker": true,
14
+ "safety_checker": [
15
+ "stable_diffusion",
16
+ "StableDiffusionSafetyChecker"
17
+ ],
18
+ "scheduler": [
19
+ "diffusers",
20
+ "PNDMScheduler"
21
+ ],
22
+ "text_encoder": [
23
+ "transformers",
24
+ "CLIPTextModel"
25
+ ],
26
+ "tokenizer": [
27
+ "transformers",
28
+ "CLIPTokenizer"
29
+ ],
30
+ "unet": [
31
+ "diffusers",
32
+ "UNet2DConditionModel"
33
+ ],
34
+ "vae": [
35
+ "diffusers",
36
+ "AutoencoderKL"
37
+ ]
38
+ }
safety_checker/config.json ADDED
@@ -0,0 +1,168 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_commit_hash": "1d0c4ebf6ff58a5caecab40fa1406526bca4b5b9",
3
+ "_name_or_path": "/mnt/slurm_home/jxjiang/.cache/huggingface/hub/models--runwayml--stable-diffusion-v1-5/snapshots/1d0c4ebf6ff58a5caecab40fa1406526bca4b5b9/safety_checker",
4
+ "architectures": [
5
+ "StableDiffusionSafetyChecker"
6
+ ],
7
+ "initializer_factor": 1.0,
8
+ "logit_scale_init_value": 2.6592,
9
+ "model_type": "clip",
10
+ "projection_dim": 768,
11
+ "text_config": {
12
+ "_name_or_path": "",
13
+ "add_cross_attention": false,
14
+ "architectures": null,
15
+ "attention_dropout": 0.0,
16
+ "bad_words_ids": null,
17
+ "begin_suppress_tokens": null,
18
+ "bos_token_id": 0,
19
+ "chunk_size_feed_forward": 0,
20
+ "cross_attention_hidden_size": null,
21
+ "decoder_start_token_id": null,
22
+ "diversity_penalty": 0.0,
23
+ "do_sample": false,
24
+ "dropout": 0.0,
25
+ "early_stopping": false,
26
+ "encoder_no_repeat_ngram_size": 0,
27
+ "eos_token_id": 2,
28
+ "exponential_decay_length_penalty": null,
29
+ "finetuning_task": null,
30
+ "forced_bos_token_id": null,
31
+ "forced_eos_token_id": null,
32
+ "hidden_act": "quick_gelu",
33
+ "hidden_size": 768,
34
+ "id2label": {
35
+ "0": "LABEL_0",
36
+ "1": "LABEL_1"
37
+ },
38
+ "initializer_factor": 1.0,
39
+ "initializer_range": 0.02,
40
+ "intermediate_size": 3072,
41
+ "is_decoder": false,
42
+ "is_encoder_decoder": false,
43
+ "label2id": {
44
+ "LABEL_0": 0,
45
+ "LABEL_1": 1
46
+ },
47
+ "layer_norm_eps": 1e-05,
48
+ "length_penalty": 1.0,
49
+ "max_length": 20,
50
+ "max_position_embeddings": 77,
51
+ "min_length": 0,
52
+ "model_type": "clip_text_model",
53
+ "no_repeat_ngram_size": 0,
54
+ "num_attention_heads": 12,
55
+ "num_beam_groups": 1,
56
+ "num_beams": 1,
57
+ "num_hidden_layers": 12,
58
+ "num_return_sequences": 1,
59
+ "output_attentions": false,
60
+ "output_hidden_states": false,
61
+ "output_scores": false,
62
+ "pad_token_id": 1,
63
+ "prefix": null,
64
+ "problem_type": null,
65
+ "projection_dim": 512,
66
+ "pruned_heads": {},
67
+ "remove_invalid_values": false,
68
+ "repetition_penalty": 1.0,
69
+ "return_dict": true,
70
+ "return_dict_in_generate": false,
71
+ "sep_token_id": null,
72
+ "suppress_tokens": null,
73
+ "task_specific_params": null,
74
+ "temperature": 1.0,
75
+ "tf_legacy_loss": false,
76
+ "tie_encoder_decoder": false,
77
+ "tie_word_embeddings": true,
78
+ "tokenizer_class": null,
79
+ "top_k": 50,
80
+ "top_p": 1.0,
81
+ "torch_dtype": null,
82
+ "torchscript": false,
83
+ "transformers_version": "4.29.2",
84
+ "typical_p": 1.0,
85
+ "use_bfloat16": false,
86
+ "vocab_size": 49408
87
+ },
88
+ "torch_dtype": "float32",
89
+ "transformers_version": null,
90
+ "vision_config": {
91
+ "_name_or_path": "",
92
+ "add_cross_attention": false,
93
+ "architectures": null,
94
+ "attention_dropout": 0.0,
95
+ "bad_words_ids": null,
96
+ "begin_suppress_tokens": null,
97
+ "bos_token_id": null,
98
+ "chunk_size_feed_forward": 0,
99
+ "cross_attention_hidden_size": null,
100
+ "decoder_start_token_id": null,
101
+ "diversity_penalty": 0.0,
102
+ "do_sample": false,
103
+ "dropout": 0.0,
104
+ "early_stopping": false,
105
+ "encoder_no_repeat_ngram_size": 0,
106
+ "eos_token_id": null,
107
+ "exponential_decay_length_penalty": null,
108
+ "finetuning_task": null,
109
+ "forced_bos_token_id": null,
110
+ "forced_eos_token_id": null,
111
+ "hidden_act": "quick_gelu",
112
+ "hidden_size": 1024,
113
+ "id2label": {
114
+ "0": "LABEL_0",
115
+ "1": "LABEL_1"
116
+ },
117
+ "image_size": 224,
118
+ "initializer_factor": 1.0,
119
+ "initializer_range": 0.02,
120
+ "intermediate_size": 4096,
121
+ "is_decoder": false,
122
+ "is_encoder_decoder": false,
123
+ "label2id": {
124
+ "LABEL_0": 0,
125
+ "LABEL_1": 1
126
+ },
127
+ "layer_norm_eps": 1e-05,
128
+ "length_penalty": 1.0,
129
+ "max_length": 20,
130
+ "min_length": 0,
131
+ "model_type": "clip_vision_model",
132
+ "no_repeat_ngram_size": 0,
133
+ "num_attention_heads": 16,
134
+ "num_beam_groups": 1,
135
+ "num_beams": 1,
136
+ "num_channels": 3,
137
+ "num_hidden_layers": 24,
138
+ "num_return_sequences": 1,
139
+ "output_attentions": false,
140
+ "output_hidden_states": false,
141
+ "output_scores": false,
142
+ "pad_token_id": null,
143
+ "patch_size": 14,
144
+ "prefix": null,
145
+ "problem_type": null,
146
+ "projection_dim": 512,
147
+ "pruned_heads": {},
148
+ "remove_invalid_values": false,
149
+ "repetition_penalty": 1.0,
150
+ "return_dict": true,
151
+ "return_dict_in_generate": false,
152
+ "sep_token_id": null,
153
+ "suppress_tokens": null,
154
+ "task_specific_params": null,
155
+ "temperature": 1.0,
156
+ "tf_legacy_loss": false,
157
+ "tie_encoder_decoder": false,
158
+ "tie_word_embeddings": true,
159
+ "tokenizer_class": null,
160
+ "top_k": 50,
161
+ "top_p": 1.0,
162
+ "torch_dtype": null,
163
+ "torchscript": false,
164
+ "transformers_version": "4.29.2",
165
+ "typical_p": 1.0,
166
+ "use_bfloat16": false
167
+ }
168
+ }
safety_checker/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:11cfe53105625af8c00faac32a430626641cce686454f3c39d837f14397d858b
3
+ size 1215981832
scheduler/scheduler_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "PNDMScheduler",
3
+ "_diffusers_version": "0.27.0.dev0",
4
+ "beta_end": 0.012,
5
+ "beta_schedule": "scaled_linear",
6
+ "beta_start": 0.00085,
7
+ "clip_sample": false,
8
+ "num_train_timesteps": 1000,
9
+ "prediction_type": "epsilon",
10
+ "set_alpha_to_one": false,
11
+ "skip_prk_steps": true,
12
+ "steps_offset": 1,
13
+ "timestep_spacing": "leading",
14
+ "trained_betas": null
15
+ }