thomasdavidwang commited on
Commit
589e19b
·
1 Parent(s): a479b57

End of training

Browse files
README.md CHANGED
@@ -19,4 +19,4 @@ You can find some example images in the following.
19
 
20
 
21
 
22
- DreamBooth for the text encoder was enabled: False.
 
19
 
20
 
21
 
22
+ DreamBooth for the text encoder was enabled: True.
checkpoint-500/optimizer.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:50b9a1c30153143b67581ae7097897f7448c926c71d34310499a397ee7578c31
3
- size 6876749715
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:10012102f26c9d062ca0290954f8458a2238cbf8cbb1afac2540385d12da22b8
3
+ size 1972597985
checkpoint-500/random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d99ef23b67f270fbf53f74136ce3fb26957f86c9177d9045d9bbe34505adcb06
3
- size 14663
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d33a8e4d1803ae27c13d3eb8765765988e54d853a7240b3b7050bf1053fb7f5d
3
+ size 14727
checkpoint-500/scheduler.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6be622d51c3bcf8a1991c02b9778f31d7e3a79a8b22e03e81a36e25205ef4c5b
3
  size 563
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f86d1f18eb9c3519789dd022870fca925dee131f3e73e8a3e37fd52f259592f4
3
  size 563
checkpoint-500/text_encoder/config.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "CompVis/stable-diffusion-v1-4",
3
+ "architectures": [
4
+ "CLIPTextModel"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 0,
8
+ "dropout": 0.0,
9
+ "eos_token_id": 2,
10
+ "hidden_act": "quick_gelu",
11
+ "hidden_size": 768,
12
+ "initializer_factor": 1.0,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 3072,
15
+ "layer_norm_eps": 1e-05,
16
+ "max_position_embeddings": 77,
17
+ "model_type": "clip_text_model",
18
+ "num_attention_heads": 12,
19
+ "num_hidden_layers": 12,
20
+ "pad_token_id": 1,
21
+ "projection_dim": 512,
22
+ "torch_dtype": "float32",
23
+ "transformers_version": "4.34.0",
24
+ "vocab_size": 49408
25
+ }
checkpoint-500/text_encoder/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:75120af77f9fc8f7284806084a909be71c449e221bb358ae089230d18c87620b
3
+ size 492308829
checkpoint-500/unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:650152d07b57c3ca56e527586f0e64592164f4efd770d8c9bdf10fff1e9f95a8
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f92653e669e0e0a6f6e7410884a0c6affca477ca18b010a07d5c5d11ff8503a8
3
  size 3438167536
logs/dreambooth/1697348685.2259984/events.out.tfevents.1697348685.3a9c9c5de5d2.77831.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:423cf63a93949478d08232cfbd93f761dcd94afd6b5d8391578bb481aa08d89e
3
+ size 2841
logs/dreambooth/1697348685.2279012/hparams.yml ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ class_data_dir: ./man
10
+ class_labels_conditioning: null
11
+ class_prompt: a photo of man
12
+ dataloader_num_workers: 0
13
+ enable_xformers_memory_efficient_attention: false
14
+ gradient_accumulation_steps: 1
15
+ gradient_checkpointing: true
16
+ hub_model_id: null
17
+ hub_token: null
18
+ instance_data_dir: ./ryan
19
+ instance_prompt: a photo of sks man
20
+ learning_rate: 2.0e-06
21
+ local_rank: -1
22
+ logging_dir: logs
23
+ lr_num_cycles: 1
24
+ lr_power: 1.0
25
+ lr_scheduler: constant
26
+ lr_warmup_steps: 0
27
+ max_grad_norm: 1.0
28
+ max_train_steps: 800
29
+ mixed_precision: null
30
+ num_class_images: 200
31
+ num_train_epochs: 4
32
+ num_validation_images: 4
33
+ offset_noise: false
34
+ output_dir: ./out
35
+ pre_compute_text_embeddings: false
36
+ pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
37
+ prior_generation_precision: null
38
+ prior_loss_weight: 1.0
39
+ push_to_hub: true
40
+ report_to: tensorboard
41
+ resolution: 512
42
+ resume_from_checkpoint: null
43
+ revision: null
44
+ sample_batch_size: 4
45
+ scale_lr: false
46
+ seed: null
47
+ set_grads_to_none: false
48
+ skip_save_text_encoder: false
49
+ snr_gamma: null
50
+ text_encoder_use_attention_mask: false
51
+ tokenizer_max_length: null
52
+ tokenizer_name: null
53
+ train_batch_size: 1
54
+ train_text_encoder: true
55
+ use_8bit_adam: true
56
+ validation_prompt: null
57
+ validation_scheduler: DPMSolverMultistepScheduler
58
+ validation_steps: 100
59
+ with_prior_preservation: true
logs/dreambooth/events.out.tfevents.1697348685.3a9c9c5de5d2.77831.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:662984c0e4bd648f55ae20f8c7efebcf18266ac6dce88596c22888419b2320d1
3
+ size 67034
text_encoder/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:778d02eb9e707c3fbaae0b67b79ea0d1399b52e624fb634f2f19375ae7c047c3
3
  size 492265168
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb0d3b69d2c6b58e31732a0e48dab7f1fcb697cb25e6c4a230ca50bca12aef27
3
  size 492265168
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1d05585ae658e7709d6cf725082e7c6b0100f05fa68bf265c8606a25df090d5e
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e276b5e9a914b594f42d5493fe5bad0c74588c531fe4d125a7560c7d1bad5900
3
  size 3438167536