PablitoGil14 commited on
Commit
57541ae
·
verified ·
1 Parent(s): 4bcd58a

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. .gitattributes +4 -0
  2. README.md +47 -0
  3. checkpoint-100/optimizer.bin +3 -0
  4. checkpoint-100/pytorch_lora_weights.safetensors +3 -0
  5. checkpoint-100/random_states_0.pkl +3 -0
  6. checkpoint-100/scheduler.bin +3 -0
  7. checkpoint-1000/optimizer.bin +3 -0
  8. checkpoint-1000/pytorch_lora_weights.safetensors +3 -0
  9. checkpoint-1000/random_states_0.pkl +3 -0
  10. checkpoint-1000/scheduler.bin +3 -0
  11. checkpoint-200/optimizer.bin +3 -0
  12. checkpoint-200/pytorch_lora_weights.safetensors +3 -0
  13. checkpoint-200/random_states_0.pkl +3 -0
  14. checkpoint-200/scheduler.bin +3 -0
  15. checkpoint-300/optimizer.bin +3 -0
  16. checkpoint-300/pytorch_lora_weights.safetensors +3 -0
  17. checkpoint-300/random_states_0.pkl +3 -0
  18. checkpoint-300/scheduler.bin +3 -0
  19. checkpoint-400/optimizer.bin +3 -0
  20. checkpoint-400/pytorch_lora_weights.safetensors +3 -0
  21. checkpoint-400/random_states_0.pkl +3 -0
  22. checkpoint-400/scheduler.bin +3 -0
  23. checkpoint-500/optimizer.bin +3 -0
  24. checkpoint-500/pytorch_lora_weights.safetensors +3 -0
  25. checkpoint-500/random_states_0.pkl +3 -0
  26. checkpoint-500/scheduler.bin +3 -0
  27. checkpoint-600/optimizer.bin +3 -0
  28. checkpoint-600/pytorch_lora_weights.safetensors +3 -0
  29. checkpoint-600/random_states_0.pkl +3 -0
  30. checkpoint-600/scheduler.bin +3 -0
  31. checkpoint-700/optimizer.bin +3 -0
  32. checkpoint-700/pytorch_lora_weights.safetensors +3 -0
  33. checkpoint-700/random_states_0.pkl +3 -0
  34. checkpoint-700/scheduler.bin +3 -0
  35. checkpoint-800/optimizer.bin +3 -0
  36. checkpoint-800/pytorch_lora_weights.safetensors +3 -0
  37. checkpoint-800/random_states_0.pkl +3 -0
  38. checkpoint-800/scheduler.bin +3 -0
  39. checkpoint-900/optimizer.bin +3 -0
  40. checkpoint-900/pytorch_lora_weights.safetensors +3 -0
  41. checkpoint-900/random_states_0.pkl +3 -0
  42. checkpoint-900/scheduler.bin +3 -0
  43. image_0.png +3 -0
  44. image_1.png +3 -0
  45. image_2.png +3 -0
  46. image_3.png +3 -0
  47. logs/dreambooth-lora/1747068032.8320847/events.out.tfevents.1747068032.50906eb71302.3723.1 +3 -0
  48. logs/dreambooth-lora/1747068032.8336604/hparams.yml +58 -0
  49. logs/dreambooth-lora/1747068265.931918/events.out.tfevents.1747068265.50906eb71302.4799.1 +3 -0
  50. logs/dreambooth-lora/1747068265.9335506/hparams.yml +58 -0
.gitattributes CHANGED
@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ image_0.png filter=lfs diff=lfs merge=lfs -text
37
+ image_1.png filter=lfs diff=lfs merge=lfs -text
38
+ image_2.png filter=lfs diff=lfs merge=lfs -text
39
+ image_3.png filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: stabilityai/stable-diffusion-2
3
+ library_name: diffusers
4
+ license: creativeml-openrail-m
5
+ inference: true
6
+ instance_prompt: a photo of a pig head
7
+ tags:
8
+ - text-to-image
9
+ - diffusers
10
+ - lora
11
+ - diffusers-training
12
+ - stable-diffusion
13
+ - stable-diffusion-diffusers
14
+ ---
15
+
16
+ <!-- This model card has been generated automatically according to the information the training script had access to. You
17
+ should probably proofread and complete it, then remove this comment. -->
18
+
19
+
20
+ # LoRA DreamBooth - PablitoGil14/mhead_pigs-stable-diffusion-v2
21
+
22
+ These are LoRA adaption weights for stabilityai/stable-diffusion-2. The weights were trained on a photo of a pig head using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following.
23
+
24
+ ![img_0](./image_0.png)
25
+ ![img_1](./image_1.png)
26
+ ![img_2](./image_2.png)
27
+ ![img_3](./image_3.png)
28
+
29
+
30
+ LoRA for the text encoder was enabled: False.
31
+
32
+
33
+ ## Intended uses & limitations
34
+
35
+ #### How to use
36
+
37
+ ```python
38
+ # TODO: add an example code snippet for running this diffusion pipeline
39
+ ```
40
+
41
+ #### Limitations and bias
42
+
43
+ [TODO: provide examples of latent issues and potential remediations]
44
+
45
+ ## Training details
46
+
47
+ [TODO: describe the data used to train the model]
checkpoint-100/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:951998e32505426374108d09da1e43434e5270047bc327c301ba72cffef81a73
3
+ size 6854266
checkpoint-100/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:075f3e8f2320d1442bea18972610e6b27d2e6feb46ff7aa2482c6a97c94fdf3a
3
+ size 3357296
checkpoint-100/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d5e7b35efff05cb972704432062712f8256a9d923003224818ff66655341a599
3
+ size 14344
checkpoint-100/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:775169a85d4b90dd73a828433c824604dc39dbd36cf213d90c3e0c82687b3460
3
+ size 1000
checkpoint-1000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd50b484c0c2d21a98af759a4709f977e5bbd300438c9c239a5d80ecef35194c
3
+ size 6854266
checkpoint-1000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0a67771a1bee37c8b528519871a4072869cc7795f1bdee80c5f2db0c2334e219
3
+ size 3357296
checkpoint-1000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f224f9791b08c4e4a9970fbf59799dda03550e7b095f1e47b6af9df43c968f9c
3
+ size 14344
checkpoint-1000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e1ef945fb67548939f1340070803ddc6f91925dd6d43f8c15ba25d113f03c5ed
3
+ size 1000
checkpoint-200/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:80516e9aeffe42446f997a7bc8948cf55f0fbf4626c34abe037a11285124ddc2
3
+ size 6854266
checkpoint-200/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ef18548540f77c0cf8eedc34104bf4f29670d8c2abdf85287f2f91efdc7f540c
3
+ size 3357296
checkpoint-200/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4094c5d2bf0d395baa68078a9c24ac6eb38197659a64b05bd0fe7accc2d1940d
3
+ size 14344
checkpoint-200/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e440cace6589de6c72ad2cf84b033c31b67b56f5f593fdfc467c84dd99ca645
3
+ size 1000
checkpoint-300/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3cbf1a2989c58fc87226ef7081ffcfad20d45f798e790c0bfde0da14e8b3789
3
+ size 6854266
checkpoint-300/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:76bcf1a4ebe793b15d9d9a74e5a386095e8960aba99093f4d555aab0497889af
3
+ size 3357296
checkpoint-300/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:40be3ebd158fcc44d338ba78d12f1678fa94a97733048558f64f1f191600f0cf
3
+ size 14344
checkpoint-300/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:78866eeca7df97c7ddfcd69f7a9a96ec8f478309052c05ea1972f881237cb751
3
+ size 1000
checkpoint-400/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:86c1f8dce7a90aef7efd42f790d29596e3a6581e217e71871abd303a0d9cf8c5
3
+ size 6854266
checkpoint-400/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dcfd8b2d3f8e98578d5cac64c33ff728ffe363a39a7c4f00dc37bdc39dfc813e
3
+ size 3357296
checkpoint-400/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:222cbe28112cb0131c244f6106bb4ed5d4ff78aadb7b9c05df034fb75a57ce8b
3
+ size 14344
checkpoint-400/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a30fb50df034f3b2fe35529f75f39a19995d00a1e34b1527e20eb794752d2f77
3
+ size 1000
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d6547a1b7c70ce0bab166914d3f0db208175ca705c44c0e1c669d47a0b9ca3e1
3
+ size 6854266
checkpoint-500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:988ed2659fb59c6b900fab8332e19eb7b271a0feabfc1708b5cb39f13b0a56f5
3
+ size 3357296
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67684ae3d0c04ffd61fa048836ba28375064271b8e1906c05fb26c2bf9f2493f
3
+ size 14344
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6cf34ac8dd2addc454aeb1ee4eaba3267c2841857e117be40ac2ed02cf581c9c
3
+ size 1000
checkpoint-600/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15d148b3df4f681edc97270e27a8186727233ba42e512a99cb614cc7725a7f0c
3
+ size 6854266
checkpoint-600/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:07bc79a6b9324efb20332664dc63e422c39c6e304f0d5b238ce60bde5fd8b79c
3
+ size 3357296
checkpoint-600/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:34c9197566d05dc65d6b254de4f8d15c8d5353b6a45d840b4a8227822807834a
3
+ size 14344
checkpoint-600/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d925ef3a312cf9f1dd743e06beceb9401782b525caeac7539c0f63bd21821f45
3
+ size 1000
checkpoint-700/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bb414f7c4a3fe3d3cd82010296dc76cef917906fe516d0d3d81d5b0eceefaab0
3
+ size 6854266
checkpoint-700/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb833f27f74cd6102a5297eb4ac071fc1961432108c50e12ef9cd05c76beb664
3
+ size 3357296
checkpoint-700/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d1a24bbb29f908ceab9aff367af267dcdfc2418f489b28b20cf954e3b562d7be
3
+ size 14344
checkpoint-700/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fb79e5bef4a7fd65954e78ff4004f852f7944a70d4c3bb940e2f9da23a8db704
3
+ size 1000
checkpoint-800/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0616f06d1d9ada4b96b4d314947dda5a09d68d6a845c3873d7b0fede5beabd90
3
+ size 6854266
checkpoint-800/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f7ab6bdf93ad0d7c0e7d347f526beb89bc8e2f59c06f0177202a90e969a1869
3
+ size 3357296
checkpoint-800/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d2b8d779a395345dffde8f537c6461fca7921536571634490b25babe63e2d904
3
+ size 14344
checkpoint-800/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e861a7df1fa5e3fd4155b5733a85f3ed8481295ab34df55c68d1a5949e2127e
3
+ size 1000
checkpoint-900/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a2b3a1ea5c776604818318b396633ece22f8bcf04cc0982603a29bdcfcebcb7e
3
+ size 6854266
checkpoint-900/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf00108275f2394a8d313560e178937af53105f3ad9e2cfdbc1a7e6a22bd3647
3
+ size 3357296
checkpoint-900/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9dc6e19d368f4b21935cd3d5047d34a23d871758fa10870aff4de1d97bc3f6a5
3
+ size 14344
checkpoint-900/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bbea3467c67a6f84349fe849eb9692b13da73d777ac16480fca3a8223d356881
3
+ size 1000
image_0.png ADDED

Git LFS Details

  • SHA256: ea0699cd6e54b6bbb5ebc6bbb41108483cf2ec165f17730790be823abfd3fc67
  • Pointer size: 131 Bytes
  • Size of remote file: 782 kB
image_1.png ADDED

Git LFS Details

  • SHA256: de0c4f40a2699bc2f22fd4a41f9b10d0e90c3c0b4b8f5c2cdba4523879bded28
  • Pointer size: 131 Bytes
  • Size of remote file: 753 kB
image_2.png ADDED

Git LFS Details

  • SHA256: 3fa4bd05c1651787fa90e98a8cb6e84ba4e09f55fd39baea8255d04de8928789
  • Pointer size: 131 Bytes
  • Size of remote file: 749 kB
image_3.png ADDED

Git LFS Details

  • SHA256: 90e31ffe0a7db95aa0a404fea68b3e98107ecf7f21c6980a5c314e8aaca4fe20
  • Pointer size: 131 Bytes
  • Size of remote file: 726 kB
logs/dreambooth-lora/1747068032.8320847/events.out.tfevents.1747068032.50906eb71302.3723.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:76c6d68882205a881f0afc95b1d74cfd5b2a42bd90bf2b61af27be3b27f117f8
3
+ size 2831
logs/dreambooth-lora/1747068032.8336604/hparams.yml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 100
8
+ checkpoints_total_limit: null
9
+ class_data_dir: null
10
+ class_labels_conditioning: null
11
+ class_prompt: null
12
+ dataloader_num_workers: 0
13
+ enable_xformers_memory_efficient_attention: false
14
+ gradient_accumulation_steps: 1
15
+ gradient_checkpointing: false
16
+ hub_model_id: null
17
+ hub_token: null
18
+ image_interpolation_mode: lanczos
19
+ instance_data_dir: mhead_pigs
20
+ instance_prompt: a photo of a pig head
21
+ learning_rate: 0.0001
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lora_dropout: 0.0
25
+ lr_num_cycles: 1
26
+ lr_power: 1.0
27
+ lr_scheduler: constant
28
+ lr_warmup_steps: 0
29
+ max_grad_norm: 1.0
30
+ max_train_steps: 1000
31
+ mixed_precision: fp16
32
+ num_class_images: 100
33
+ num_train_epochs: 200
34
+ num_validation_images: 4
35
+ output_dir: mhead_pigs-stable-diffusion-v2
36
+ pre_compute_text_embeddings: false
37
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-2
38
+ prior_generation_precision: null
39
+ prior_loss_weight: 1.0
40
+ push_to_hub: true
41
+ rank: 4
42
+ report_to: tensorboard
43
+ resolution: 768
44
+ resume_from_checkpoint: null
45
+ revision: null
46
+ sample_batch_size: 4
47
+ scale_lr: false
48
+ seed: 0
49
+ text_encoder_use_attention_mask: false
50
+ tokenizer_max_length: null
51
+ tokenizer_name: null
52
+ train_batch_size: 1
53
+ train_text_encoder: false
54
+ use_8bit_adam: false
55
+ validation_epochs: 50
56
+ validation_prompt: a photo of human with a pig head
57
+ variant: null
58
+ with_prior_preservation: false
logs/dreambooth-lora/1747068265.931918/events.out.tfevents.1747068265.50906eb71302.4799.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:74f14263a90b71aaac1cb0010f05a9a8ad83844696600709c821bceda93a261c
3
+ size 2831
logs/dreambooth-lora/1747068265.9335506/hparams.yml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 100
8
+ checkpoints_total_limit: null
9
+ class_data_dir: null
10
+ class_labels_conditioning: null
11
+ class_prompt: null
12
+ dataloader_num_workers: 0
13
+ enable_xformers_memory_efficient_attention: false
14
+ gradient_accumulation_steps: 1
15
+ gradient_checkpointing: false
16
+ hub_model_id: null
17
+ hub_token: null
18
+ image_interpolation_mode: lanczos
19
+ instance_data_dir: mhead_pigs
20
+ instance_prompt: a photo of a pig head
21
+ learning_rate: 0.0001
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lora_dropout: 0.0
25
+ lr_num_cycles: 1
26
+ lr_power: 1.0
27
+ lr_scheduler: constant
28
+ lr_warmup_steps: 0
29
+ max_grad_norm: 1.0
30
+ max_train_steps: 1000
31
+ mixed_precision: fp16
32
+ num_class_images: 100
33
+ num_train_epochs: 200
34
+ num_validation_images: 4
35
+ output_dir: mhead_pigs-stable-diffusion-v2
36
+ pre_compute_text_embeddings: false
37
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-2
38
+ prior_generation_precision: null
39
+ prior_loss_weight: 1.0
40
+ push_to_hub: true
41
+ rank: 4
42
+ report_to: tensorboard
43
+ resolution: 768
44
+ resume_from_checkpoint: null
45
+ revision: null
46
+ sample_batch_size: 4
47
+ scale_lr: false
48
+ seed: 0
49
+ text_encoder_use_attention_mask: false
50
+ tokenizer_max_length: null
51
+ tokenizer_name: null
52
+ train_batch_size: 1
53
+ train_text_encoder: false
54
+ use_8bit_adam: false
55
+ validation_epochs: 50
56
+ validation_prompt: a photo of human with a pig head
57
+ variant: null
58
+ with_prior_preservation: false