End of training

README.md

@@ -0,0 +1,21 @@
+
+---
+license: creativeml-openrail-m
+base_model: runwayml/stable-diffusion-v1-5
+tags:
+- stable-diffusion
+- stable-diffusion-diffusers
+- text-to-image
+- diffusers
+- lora
+inference: true
+---
+    
+# LoRA text2image fine-tuning - umaru97/flickr30k-lora
+These are LoRA adaption weights for runwayml/stable-diffusion-v1-5. The weights were fine-tuned on the umaru97/flickr30k_train_val_test dataset. You can find some example images in the following. 
+
+![img_0](./image_0.png)
+![img_1](./image_1.png)
+![img_2](./image_2.png)
+![img_3](./image_3.png)
+

checkpoint-2000/optimizer.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:31422443d598b2b008dbf08ff0817a33753798a2b868a8ab924557a5190c0c0a
+size 6591685

checkpoint-2000/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f1af87a5b5db554b2572ecec3e9c43ebc4ca4ed449de840bc2aadccb54e5dad1
+size 3285965

checkpoint-2000/random_states_0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d6e1d9ed29fe4a374a20496c1815e5f5541e9ebeb4c1c161ad47309059cd09ce
+size 14663

checkpoint-2000/scaler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dd2de9749828adacdf103bf6e9592702bb7067a2c1df27dd62ab38c1eb8c070f
+size 557

checkpoint-2000/scheduler.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cfb08ff088790178fb856390b7fafbc19d29b343d2e466e0883e97f75aaa2b6b
+size 563

logs/text2image-fine-tune/1691132013.450615/events.out.tfevents.1691132013.b4ff7e718147.19907.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e48ab5154d65abfe1a969e56c1a88b7f25396a3cf659231b7995d203f1550bf0
+size 2333

logs/text2image-fine-tune/1691132013.4635336/hparams.yml

@@ -0,0 +1,49 @@
+adam_beta1: 0.9
+adam_beta2: 0.999
+adam_epsilon: 1.0e-08
+adam_weight_decay: 0.01
+allow_tf32: false
+cache_dir: null
+caption_column: caption
+center_crop: true
+checkpointing_steps: 2000
+checkpoints_total_limit: null
+dataloader_num_workers: 8
+dataset_config_name: null
+dataset_name: umaru97/flickr30k_train_val_test
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 4
+gradient_checkpointing: false
+hub_model_id: flickr30k-lora
+hub_token: null
+image_column: image
+learning_rate: 0.0001
+local_rank: -1
+logging_dir: logs
+lr_scheduler: cosine
+lr_warmup_steps: 0
+max_grad_norm: 1.0
+max_train_samples: null
+max_train_steps: 1813
+mixed_precision: null
+noise_offset: 0
+num_train_epochs: 1
+num_validation_images: 4
+output_dir: /content/drive/My Drive/interview/sembcorp/takehome_assignment/task3/models
+prediction_type: null
+pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
+push_to_hub: true
+random_flip: true
+rank: 4
+report_to: tensorboard
+resolution: 512
+resume_from_checkpoint: null
+revision: null
+scale_lr: false
+seed: 77
+snr_gamma: null
+train_batch_size: 4
+train_data_dir: null
+use_8bit_adam: false
+validation_epochs: 1
+validation_prompt: Cat

logs/text2image-fine-tune/1691132105.9168317/events.out.tfevents.1691132105.b4ff7e718147.20384.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a3e98b37a9b0897b1c4dbc5c13cee73f78f4d5ac44a7d689a4ba0c8254d84f83
+size 2333

logs/text2image-fine-tune/1691132105.9292111/hparams.yml

@@ -0,0 +1,49 @@
+adam_beta1: 0.9
+adam_beta2: 0.999
+adam_epsilon: 1.0e-08
+adam_weight_decay: 0.01
+allow_tf32: false
+cache_dir: null
+caption_column: caption
+center_crop: true
+checkpointing_steps: 2000
+checkpoints_total_limit: null
+dataloader_num_workers: 8
+dataset_config_name: null
+dataset_name: umaru97/flickr30k_train_val_test
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 4
+gradient_checkpointing: false
+hub_model_id: flickr30k-lora
+hub_token: null
+image_column: image
+learning_rate: 0.0001
+local_rank: -1
+logging_dir: logs
+lr_scheduler: cosine
+lr_warmup_steps: 0
+max_grad_norm: 1.0
+max_train_samples: null
+max_train_steps: 3626
+mixed_precision: null
+noise_offset: 0
+num_train_epochs: 2
+num_validation_images: 4
+output_dir: /content/drive/My Drive/interview/sembcorp/takehome_assignment/task3/models
+prediction_type: null
+pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
+push_to_hub: true
+random_flip: true
+rank: 4
+report_to: tensorboard
+resolution: 512
+resume_from_checkpoint: null
+revision: null
+scale_lr: false
+seed: 77
+snr_gamma: null
+train_batch_size: 4
+train_data_dir: null
+use_8bit_adam: false
+validation_epochs: 1
+validation_prompt: Cat

logs/text2image-fine-tune/events.out.tfevents.1691132013.b4ff7e718147.19907.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b8e54542caafd3b4d17c5dedd94fa17f58c476916c11f244f0c207e6e4bef334
+size 616

logs/text2image-fine-tune/events.out.tfevents.1691132105.b4ff7e718147.20384.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7fe2bee1e1a7b1676d494a2d71e3cd6c7593a5b9b11c53663c6550bdda1596ad
+size 4148192

pytorch_lora_weights.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4a17d470526debf403bba6a8e2ece1032352af2df8a90068c061d6bfb88eb473
+size 3287771