briannlongzhao commited on
Commit
31685a8
1 Parent(s): 5ea5bea

End of training

Browse files
<new1>.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8c778df31ec2cca3ed6b9b999532b93980eadb5468fcaee1221c68abce33a388
3
  size 202379944
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be41875e05703d53f65ef1322de05e5b7e955d2fc119f6cea5e3ce942cb48e7a
3
  size 202379944
README.md CHANGED
@@ -2,21 +2,21 @@
2
  ---
3
  license: creativeml-openrail-m
4
  base_model: stabilityai/stable-diffusion-2-1
5
- instance_prompt: a photo of sks football helmet
6
  tags:
7
  - stable-diffusion
8
  - stable-diffusion-diffusers
9
  - text-to-image
10
  - diffusers
11
- - dreambooth
12
  inference: true
13
  ---
14
 
15
- # DreamBooth - briannlongzhao/56
16
 
17
- This is a dreambooth model derived from stabilityai/stable-diffusion-2-1. The weights were trained on a photo of sks football helmet using [DreamBooth](https://dreambooth.github.io/).
18
- You can find some example images in the following.
19
 
20
 
21
 
22
- DreamBooth for the text encoder was enabled: False.
 
 
2
  ---
3
  license: creativeml-openrail-m
4
  base_model: stabilityai/stable-diffusion-2-1
5
+ instance_prompt: a photo of <new1> football helmet
6
  tags:
7
  - stable-diffusion
8
  - stable-diffusion-diffusers
9
  - text-to-image
10
  - diffusers
11
+ - custom-diffusion
12
  inference: true
13
  ---
14
 
15
+ # Custom Diffusion - briannlongzhao/56
16
 
17
+ These are Custom Diffusion adaption weights for stabilityai/stable-diffusion-2-1. The weights were trained on a photo of <new1> football helmet using [Custom Diffusion](https://www.cs.cmu.edu/~custom-diffusion). You can find some example images in the following.
 
18
 
19
 
20
 
21
+
22
+ For more details on the training, please follow [this link](https://github.com/huggingface/diffusers/blob/main/examples/custom_diffusion).
logs/custom-diffusion/1706459269.9749053/events.out.tfevents.1706459270.node-0.2733.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a59da59e99b70ba18d17431ff9affbe2cd38149eee786c583e7e311fcc8d7a84
3
+ size 2812
logs/custom-diffusion/1706459270.3798404/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: 250
8
+ checkpoints_total_limit: null
9
+ class_data_dir: null
10
+ class_name: football helmet
11
+ class_prompt: null
12
+ concepts_list: null
13
+ dataloader_num_workers: 2
14
+ enable_xformers_memory_efficient_attention: false
15
+ freeze_model: crossattn_kv
16
+ gradient_accumulation_steps: 1
17
+ gradient_checkpointing: true
18
+ hflip: true
19
+ hub_model_id: null
20
+ hub_token: null
21
+ idx: 56
22
+ initializer_token: ktn+pll+ucd
23
+ instance_data_dir: /mnt/default/imagenet/images/train_medium/n03379051
24
+ instance_prompt: a photo of <new1> football helmet
25
+ learning_rate: 1.0e-05
26
+ local_rank: -1
27
+ logging_dir: logs
28
+ lr_scheduler: constant
29
+ lr_warmup_steps: 0
30
+ max_grad_norm: 1.0
31
+ max_train_steps: null
32
+ mixed_precision: fp16
33
+ modifier_token: <new1>
34
+ no_safe_serialization: true
35
+ noaug: false
36
+ num_class_images: 200
37
+ num_train_epochs: 5
38
+ num_validation_images: 2
39
+ output_dir: /mnt/default/custom_diffusion_models_medium/56
40
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-2-1
41
+ prior_generation_precision: null
42
+ prior_loss_weight: 1.0
43
+ push_to_hub: true
44
+ real_prior: false
45
+ report_to: tensorboard
46
+ resolution: 512
47
+ resume_from_checkpoint: null
48
+ revision: null
49
+ sample_batch_size: 4
50
+ scale_lr: true
51
+ seed: 42
52
+ set_grads_to_none: false
53
+ tokenizer_name: null
54
+ train_batch_size: 4
55
+ use_8bit_adam: true
56
+ validation_prompt: null
57
+ validation_steps: 50
58
+ with_prior_preservation: false
59
+ wnid: n03379051
logs/custom-diffusion/events.out.tfevents.1706459269.node-0.2733.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:429798f6715caf6fc26e0ce424fbdf8b00014daa89ccd262193be8c2e21eff2a
3
+ size 10338
pytorch_custom_diffusion_weights.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:33e4af921d9b316d1ebf2429ece3eba4aca8333ad6438e4a522a63c13d67d7fb
3
  size 102250391
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce5930c5bb0a17a86b1e0d51d3c7c7d1c098bff13b348b19a72fb298c351c0a6
3
  size 102250391