dbecker1 commited on
Commit
cbf7aa8
·
1 Parent(s): bbc1d71

End of training

Browse files
README.md ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ license: creativeml-openrail-m
4
+ base_model: stabilityai/stable-diffusion-xl-base-1.0
5
+ dataset: lambdalabs/pokemon-blip-captions
6
+ tags:
7
+ - stable-diffusion-xl
8
+ - stable-diffusion-xl-diffusers
9
+ - text-to-image
10
+ - diffusers
11
+ - lora
12
+ inference: true
13
+ ---
14
+
15
+ # LoRA text2image fine-tuning - dbecker1/sd-pokemon-model-lora-sdxl
16
+
17
+ These are LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0. The weights were fine-tuned on the lambdalabs/pokemon-blip-captions dataset. You can find some example images in the following.
18
+
19
+
20
+
21
+ LoRA for the text encoder was enabled: False.
22
+
23
+ Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.
checkpoint-1000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a33bab2c0fede6705d4f989bbc87e0d767901eff1dd86b7e79bd8b4ef2448b4d
3
+ size 47392445
checkpoint-1000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:993050d7d125c7eff886b9d8256a54e6e224e5fa15aba4c0131835f3cc498f90
3
+ size 23401064
checkpoint-1000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e9cf9c331da7ed2e4053a5df8d6ee1e956bc60e9fa045a0861648258fb817a3c
3
+ size 14663
checkpoint-1000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68cff80b680ddf6e7abbef98b5f336b97f9b5963e2209307f639383870e8cc71
3
+ size 557
checkpoint-1000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:963c3085720a3a3ca1d178559a0f2dc5d5b36fa395a4663f0194bfac4a754038
3
+ size 563
checkpoint-1500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ecf64e7ce0e91d7e25d4d8c4a2666d79ada2077e70f9641cbe5c503314071cb
3
+ size 47392445
checkpoint-1500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:adb6475369f56e7240356be1dcda7375a1083cdd887d0fa686e1c37ef81974a5
3
+ size 23401064
checkpoint-1500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7e19d2f1cac2e35da233dd35a2a1a9979031d94fe4de63493193601628f50569
3
+ size 14599
checkpoint-1500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:203a72d6c29f42a0e2964fdddc8d7a98df1eccee78fea9de0fa416613390f5c6
3
+ size 557
checkpoint-1500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d6ab1339da08e3fa015ca19e5a001977bd59cee7c7573120ae85950b1c7964a0
3
+ size 563
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:29ab8f17d04faa7b7962e22de90c178bbbfcd993e0a35a9407a3efa10377128b
3
+ size 47392445
checkpoint-500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab80e2b9b7e6c6237ccd19d8ee7ba626f1c2d7f95115d36b2b1b74fcc84d9404
3
+ size 23401064
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:38abc03edfab4af92a1c504e80e4740b59f69740de642959164d858adc2dca0b
3
+ size 14663
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 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:62783ff6f0ae2c2ccc177b097e430c03940c5e70c38a191df273c5cf7ab1227c
3
+ size 563
logs/text2image-fine-tune/1694767137.496222/events.out.tfevents.1694767137.vertex-gpu-t.45623.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ef5493944549806e8d11687b43543647012f5a80fa432e5f4216824535f6dca7
3
+ size 2404
logs/text2image-fine-tune/1694767137.4977875/hparams.yml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 1666
28
+ mixed_precision: fp16
29
+ noise_offset: 0
30
+ num_train_epochs: 2
31
+ num_validation_images: 4
32
+ output_dir: sd-pokemon-model-lora-sdxl
33
+ prediction_type: null
34
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-xl-base-1.0
35
+ pretrained_vae_model_name_or_path: madebyollin/sdxl-vae-fp16-fix
36
+ push_to_hub: true
37
+ random_flip: true
38
+ rank: 4
39
+ report_to: tensorboard
40
+ resolution: 1024
41
+ resume_from_checkpoint: null
42
+ revision: null
43
+ scale_lr: false
44
+ seed: 42
45
+ snr_gamma: null
46
+ train_batch_size: 1
47
+ train_data_dir: null
48
+ train_text_encoder: false
49
+ use_8bit_adam: false
50
+ validation_epochs: 1
51
+ validation_prompt: null
logs/text2image-fine-tune/events.out.tfevents.1694767137.vertex-gpu-t.45623.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c56e8d681579c6947305ba6d8091213fecf93a20bcf04f503be2e666e9a64be
3
+ size 81595
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:665ca83ac48f15aad443e93598eb385260a53c1ff20dfd79054a70ad8b8f22f9
3
+ size 23401064