briannlongzhao commited on
Commit
def2199
·
verified ·
1 Parent(s): dcefdbc

End of training

Browse files
README.md CHANGED
@@ -2,7 +2,7 @@
2
  ---
3
  license: creativeml-openrail-m
4
  base_model: stabilityai/stable-diffusion-2-1
5
- instance_prompt: a painting of sks star night
6
  tags:
7
  - stable-diffusion
8
  - stable-diffusion-diffusers
@@ -14,7 +14,7 @@ inference: true
14
 
15
  # DreamBooth - briannlongzhao/smile_dreambooth
16
 
17
- This is a dreambooth model derived from stabilityai/stable-diffusion-2-1. The weights were trained on a painting of sks star night using [DreamBooth](https://dreambooth.github.io/).
18
  You can find some example images in the following.
19
 
20
 
 
2
  ---
3
  license: creativeml-openrail-m
4
  base_model: stabilityai/stable-diffusion-2-1
5
+ instance_prompt: a photo of a sks smile
6
  tags:
7
  - stable-diffusion
8
  - stable-diffusion-diffusers
 
14
 
15
  # DreamBooth - briannlongzhao/smile_dreambooth
16
 
17
+ This is a dreambooth model derived from stabilityai/stable-diffusion-2-1. The weights were trained on a photo of a sks smile using [DreamBooth](https://dreambooth.github.io/).
18
  You can find some example images in the following.
19
 
20
 
checkpoint-500/optimizer.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e3b526c02910cbdcea1d4b7cf4d1047b12e09b09a4e96e420dcb7a7a3af2291c
3
  size 1737914501
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:284251f2de5c2f6858a06b418a1bdc15a73f62b4cc778a159aa0d8f37fee102d
3
  size 1737914501
checkpoint-500/random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:dbf4034c70fe17e36f56ac9929d40a3a211111a00923416fa91980d333ba107d
3
- size 14727
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:242a11d072aa21d82574d7bf5803253ee21e8c40ec9730f6f3b62d16c66ee5ff
3
+ size 17683
checkpoint-500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3f196a54202bb4ba1220e8c59f42f9cda0702d68ea83147d814c2fb2f36b8f2
3
+ size 557
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:6afeb3b0b41c5a451ea879e32dbb48a3e4e665fe8057b38a556504e29915a4a3
3
  size 563
checkpoint-500/unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:15efbc44cbdf9f6da6b2766061de1db5f7701d289875c8886bef80037e3962a4
3
  size 3463726504
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3ca8fb69cc22b00c3538f2059a7e5fec273eeacd907888bfe845a066eb37b332
3
  size 3463726504
logs/dreambooth/1732583811.3699548/events.out.tfevents.1732583811.iGpu15.3134693.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31bb2e063c288b596869fea8cce570562c60a317e4eee172d0be0e75b221515b
3
+ size 3013
logs/dreambooth/1732583811.3755972/hparams.yml ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: /lab/tmpig7b/u/brian-data/eval_4/smile_preserve
10
+ class_labels_conditioning: null
11
+ class_prompt: a photo of a smile
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
+ idx: null
19
+ instance_data_dir: /lab/tmpig7b/u/brian-data/eval_4/smile/reference
20
+ instance_prompt: a photo of a sks smile
21
+ learning_rate: 5.0e-06
22
+ local_rank: 0
23
+ logging_dir: logs
24
+ lr_num_cycles: 1
25
+ lr_power: 1.0
26
+ lr_scheduler: constant
27
+ lr_warmup_steps: 0
28
+ max_grad_norm: 1.0
29
+ max_train_steps: 500
30
+ mixed_precision: fp16
31
+ num_class_images: 50
32
+ num_train_epochs: 39
33
+ num_validation_images: 4
34
+ offset_noise: false
35
+ output_dir: dreambooth_models/smile_dreambooth
36
+ pre_compute_text_embeddings: false
37
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-2-1
38
+ prior_generation_precision: null
39
+ prior_loss_weight: 1.0
40
+ push_to_hub: true
41
+ report_to: tensorboard
42
+ resolution: 768
43
+ resume_from_checkpoint: null
44
+ revision: null
45
+ sample_batch_size: 4
46
+ scale_lr: false
47
+ seed: null
48
+ set_grads_to_none: false
49
+ skip_save_text_encoder: false
50
+ snr_gamma: null
51
+ text_encoder_use_attention_mask: false
52
+ tokenizer_max_length: null
53
+ tokenizer_name: null
54
+ train_batch_size: 1
55
+ train_text_encoder: false
56
+ use_8bit_adam: true
57
+ validation_prompt: null
58
+ validation_scheduler: DPMSolverMultistepScheduler
59
+ validation_steps: 100
60
+ with_prior_preservation: true
logs/dreambooth/events.out.tfevents.1732583811.iGpu15.3134693.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:65411ceabbb9f03146dba3412844fcdc97c70b0f844fd3a5f0e606e2509b88a5
3
+ size 41834
text_encoder/config.json CHANGED
@@ -19,7 +19,7 @@
19
  "num_hidden_layers": 23,
20
  "pad_token_id": 1,
21
  "projection_dim": 512,
22
- "torch_dtype": "bfloat16",
23
  "transformers_version": "4.28.1",
24
  "vocab_size": 49408
25
  }
 
19
  "num_hidden_layers": 23,
20
  "pad_token_id": 1,
21
  "projection_dim": 512,
22
+ "torch_dtype": "float16",
23
  "transformers_version": "4.28.1",
24
  "vocab_size": 49408
25
  }
text_encoder/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:79d5659830653fd5ee1cde49e26d23772564b1f24e46188995e6a6f502373e98
3
- size 680821472
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:681c555376658c81dc273f2d737a2aeb23ddb6d1d8e5b3a7064636d359a22668
3
+ size 680821096
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e1015141eb29f444cc56680dfb4a5175eea9150be22b743552303fdd1b6c5a68
3
  size 3463726504
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3ca8fb69cc22b00c3538f2059a7e5fec273eeacd907888bfe845a066eb37b332
3
  size 3463726504
vae/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "AutoencoderKL",
3
  "_diffusers_version": "0.22.0.dev0",
4
- "_name_or_path": "/home/v-enlongzhao/.cache/huggingface/hub/models--stabilityai--stable-diffusion-2-1/snapshots/5cae40e6a2745ae2b01ad92ae5043f95f23644d6/vae",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  128,
 
1
  {
2
  "_class_name": "AutoencoderKL",
3
  "_diffusers_version": "0.22.0.dev0",
4
+ "_name_or_path": "/lab/briannlz/.cache/huggingface/hub/models--stabilityai--stable-diffusion-2-1/snapshots/5cae40e6a2745ae2b01ad92ae5043f95f23644d6/vae",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  128,