Hugging Face's logo

abwqr
/
text2img_vision

Text-to-Image
Diffusers
TensorBoard
stable-diffusion
stable-diffusion-diffusers
lora
Model card Files Metrics Community
abwqr commited on
Commit
f31017c
•
1 Parent(s): 043f79e

End of training

Browse files
Files changed (12) hide show
  1. README.md +17 -0
  2. checkpoint-500/optimizer.bin +3 -0
  3. checkpoint-500/pytorch_model.bin +3 -0
  4. checkpoint-500/random_states_0.pkl +3 -0
  5. checkpoint-500/scheduler.bin +3 -0
  6. logs/text2image-fine-tune/1689849034.3115041/events.out.tfevents.1689849034.5e88573885a6.1835.1 +3 -0
  7. logs/text2image-fine-tune/1689849034.313176/hparams.yml +49 -0
  8. logs/text2image-fine-tune/1689851106.2992063/events.out.tfevents.1689851106.5e88573885a6.25778.1 +3 -0
  9. logs/text2image-fine-tune/1689851106.3009179/hparams.yml +49 -0
  10. logs/text2image-fine-tune/events.out.tfevents.1689849034.5e88573885a6.1835.0 +3 -0
  11. logs/text2image-fine-tune/events.out.tfevents.1689851106.5e88573885a6.25778.0 +3 -0
  12. pytorch_lora_weights.bin +3 -0
README.md ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ license: creativeml-openrail-m
4
+ base_model: nota-ai/bk-sdm-small
5
+ tags:
6
+ - stable-diffusion
7
+ - stable-diffusion-diffusers
8
+ - text-to-image
9
+ - diffusers
10
+ - lora
11
+ inference: true
12
+ ---
13
+
14
+ # LoRA text2image fine-tuning - abwqr/text2img_vision
15
+ These are LoRA adaption weights for nota-ai/bk-sdm-small. The weights were fine-tuned on the abwqr/michael_scott dataset. You can find some example images in the following.
16
+
17
+
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d6fb9ae27c3043a3182e54eb2e95ae11a06e1f465cfd53eedfcd36474941390a
3
+ size 3573381
checkpoint-500/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0c131174653d43dc2657e41e73bb7fd7e24f5d392684fe0cc443fcdb1217ff9c
3
+ size 1780941
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:60568df810f6730237f5f21de23048808f6cb5b8707a7864ce357d145ea11a1c
3
+ size 14727
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab20ead000e6ae60e9a32d69fb1fb976d10dba2a065da680293a8202053a1975
3
+ size 563
logs/text2image-fine-tune/1689849034.3115041/events.out.tfevents.1689849034.5e88573885a6.1835.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba9002bbe47141a63dd0f5a1466d08e3601c7c25f6b35113916e6baecd79d513
3
+ size 2182
logs/text2image-fine-tune/1689849034.313176/hparams.yml ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: true
9
+ checkpointing_steps: 500
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 8
12
+ dataset_config_name: null
13
+ dataset_name: abwqr/michael_scott
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 4
16
+ gradient_checkpointing: true
17
+ hub_model_id: abwqr/text2img_vision
18
+ hub_token: null
19
+ image_column: image
20
+ learning_rate: 1.0e-05
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: 500
28
+ mixed_precision: null
29
+ noise_offset: 0
30
+ num_train_epochs: 500
31
+ num_validation_images: 4
32
+ output_dir: abwqr/text2img_vision
33
+ prediction_type: null
34
+ pretrained_model_name_or_path: nota-ai/bk-sdm-small
35
+ push_to_hub: true
36
+ random_flip: true
37
+ rank: 4
38
+ report_to: tensorboard
39
+ resolution: 300
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: null
44
+ snr_gamma: null
45
+ train_batch_size: 4
46
+ train_data_dir: null
47
+ use_8bit_adam: false
48
+ validation_epochs: 1
49
+ validation_prompt: null
logs/text2image-fine-tune/1689851106.2992063/events.out.tfevents.1689851106.5e88573885a6.25778.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:458d33cd15d4cdb669481a25c0b0d068129c8e2c87777d34649df35aa303caef
3
+ size 2182
logs/text2image-fine-tune/1689851106.3009179/hparams.yml ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: true
9
+ checkpointing_steps: 500
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 8
12
+ dataset_config_name: null
13
+ dataset_name: abwqr/michael_scott
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 4
16
+ gradient_checkpointing: true
17
+ hub_model_id: abwqr/text2img_vision
18
+ hub_token: null
19
+ image_column: image
20
+ learning_rate: 1.0e-05
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: 1
28
+ mixed_precision: null
29
+ noise_offset: 0
30
+ num_train_epochs: 1
31
+ num_validation_images: 4
32
+ output_dir: abwqr/text2img_vision
33
+ prediction_type: null
34
+ pretrained_model_name_or_path: nota-ai/bk-sdm-small
35
+ push_to_hub: true
36
+ random_flip: true
37
+ rank: 4
38
+ report_to: tensorboard
39
+ resolution: 300
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: null
44
+ snr_gamma: null
45
+ train_batch_size: 4
46
+ train_data_dir: null
47
+ use_8bit_adam: false
48
+ validation_epochs: 1
49
+ validation_prompt: null
logs/text2image-fine-tune/events.out.tfevents.1689849034.5e88573885a6.1835.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ceee07f73e0a5c1002a46731040e8bda0e09d21c9a50d030106c53f0953c9185
3
+ size 24461
logs/text2image-fine-tune/events.out.tfevents.1689851106.5e88573885a6.25778.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:22994b5420dd558b6205b40bb47fc8b80606443609fe33f330a0e7ed234b9ca0
3
+ size 136
pytorch_lora_weights.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ab94cdf7d7bac16ffb01341a4adb59fafb291c42a9a305abdade85e79b358ae
3
+ size 1781963