Jingya HF staff commited on
Commit
fc6ea1d
1 Parent(s): 3ac1593
README.md CHANGED
@@ -1,3 +1,5 @@
1
  ---
2
  license: apache-2.0
3
  ---
 
 
 
1
  ---
2
  license: apache-2.0
3
  ---
4
+
5
+ tiny lora trained with [pokemon](https://github.com/huggingface/diffusers/tree/main/examples/text_to_image#training-with-lora) for `hf-internal-testing/tiny-stable-diffusion-torch`. [TEST CIS ONLY]
logs/text2image-fine-tune/1708003904.7353144/events.out.tfevents.1708003904.ip-172-31-35-53.7661.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:254f821a680c729dd75179defe37f51c39a3acec68f980e2c74cee347ea470a7
3
+ size 2192
logs/text2image-fine-tune/1708003904.736912/hparams.yml ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ cache_dir: null
7
+ caption_column: text
8
+ center_crop: false
9
+ checkpointing_steps: 500
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: lambdalabs/pokemon-blip-captions
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 1
16
+ gradient_checkpointing: false
17
+ hub_model_id: null
18
+ hub_token: null
19
+ image_column: image
20
+ learning_rate: 0.0001
21
+ local_rank: -1
22
+ logging_dir: logs
23
+ lr_scheduler: constant
24
+ lr_warmup_steps: 0
25
+ max_grad_norm: 1.0
26
+ max_train_samples: null
27
+ max_train_steps: 4165
28
+ mixed_precision: null
29
+ noise_offset: 0
30
+ num_train_epochs: 5
31
+ num_validation_images: 4
32
+ output_dir: sd-pokemon-model-lora
33
+ prediction_type: null
34
+ pretrained_model_name_or_path: hf-internal-testing/tiny-stable-diffusion-torch
35
+ push_to_hub: false
36
+ random_flip: true
37
+ rank: 4
38
+ report_to: tensorboard
39
+ resolution: 64
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ snr_gamma: null
45
+ train_batch_size: 1
46
+ train_data_dir: null
47
+ use_8bit_adam: false
48
+ validation_epochs: 1
49
+ validation_prompt: null
50
+ variant: null
logs/text2image-fine-tune/1708003968.0146737/events.out.tfevents.1708003968.ip-172-31-35-53.7802.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2988b73940dfea6a03041abfb378acd2c1f898126b8ae33f0d702808ec38883d
3
+ size 2192
logs/text2image-fine-tune/1708003968.0162363/hparams.yml ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ cache_dir: null
7
+ caption_column: text
8
+ center_crop: false
9
+ checkpointing_steps: 500
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: lambdalabs/pokemon-blip-captions
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 1
16
+ gradient_checkpointing: false
17
+ hub_model_id: null
18
+ hub_token: null
19
+ image_column: image
20
+ learning_rate: 0.0001
21
+ local_rank: -1
22
+ logging_dir: logs
23
+ lr_scheduler: constant
24
+ lr_warmup_steps: 0
25
+ max_grad_norm: 1.0
26
+ max_train_samples: null
27
+ max_train_steps: 4165
28
+ mixed_precision: null
29
+ noise_offset: 0
30
+ num_train_epochs: 5
31
+ num_validation_images: 4
32
+ output_dir: sd-pokemon-model-lora
33
+ prediction_type: null
34
+ pretrained_model_name_or_path: hf-internal-testing/tiny-stable-diffusion-torch
35
+ push_to_hub: false
36
+ random_flip: true
37
+ rank: 4
38
+ report_to: tensorboard
39
+ resolution: 64
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ snr_gamma: null
45
+ train_batch_size: 1
46
+ train_data_dir: null
47
+ use_8bit_adam: false
48
+ validation_epochs: 1
49
+ validation_prompt: null
50
+ variant: null
logs/text2image-fine-tune/events.out.tfevents.1708003904.ip-172-31-35-53.7661.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cd8042081aae6bb519c194aea38f19aad3e0d8d0774c6bdec7ee446bbf0edfee
3
+ size 88
logs/text2image-fine-tune/events.out.tfevents.1708003968.ip-172-31-35-53.7802.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ef99f9f098964706428788287d26d9c12eb05c0e06e75767011f30bfc9806e2e
3
+ size 204046
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0db076ec473a9c4a2fbc89ee67fe8d0e642b67438e266f56d480d05453d11e5e
3
+ size 105744