End of training

.gitattributes

@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+image_0.png filter=lfs diff=lfs merge=lfs -text
+image_1.png filter=lfs diff=lfs merge=lfs -text
+image_2.png filter=lfs diff=lfs merge=lfs -text
+image_3.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,43 @@
+---
+base_model: Manojb/stable-diffusion-2-1-base
+library_name: diffusers
+license: creativeml-openrail-m
+inference: true
+tags:
+- stable-diffusion
+- stable-diffusion-diffusers
+- text-to-image
+- diffusers
+- diffusers-training
+- lora
+---
+
+<!-- This model card has been generated automatically according to the information the training script had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+
+# LoRA text2image fine-tuning - zhangyi617/sd2_pokemon_lora
+These are LoRA adaption weights for Manojb/stable-diffusion-2-1-base. The weights were fine-tuned on the reach-vb/pokemon-blip-captions 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)
+
+
+
+## Intended uses & limitations
+
+#### How to use
+
+```python
+# TODO: add an example code snippet for running this diffusion pipeline
+```
+
+#### Limitations and bias
+
+[TODO: provide examples of latent issues and potential remediations]
+
+## Training details
+
+[TODO: describe the data used to train the model]

checkpoint-4000/model.safetensors

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

checkpoint-4000/optimizer.bin

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

checkpoint-4000/pytorch_lora_weights.safetensors

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

checkpoint-4000/random_states_0.pkl

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

checkpoint-4000/scaler.pt

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

checkpoint-4000/scheduler.bin

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

logs/text2image-fine-tune/1771485530.0075595/events.out.tfevents.1771485530.wmgubws17.wmgds.wmg.warwick.ac.uk.3161872.1

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

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

@@ -0,0 +1,51 @@
+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: text
+center_crop: true
+checkpointing_steps: 4000
+checkpoints_total_limit: null
+dataloader_num_workers: 8
+dataset_config_name: null
+dataset_name: reach-vb/pokemon-blip-captions
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 1
+gradient_checkpointing: false
+hub_model_id: sd2_pokemon_lora
+hub_token: null
+image_column: image
+image_interpolation_mode: lanczos
+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: 4180
+mixed_precision: fp16
+noise_offset: 0
+num_train_epochs: 20
+num_validation_images: 4
+output_dir: output_models/sd2_pokemon_lora
+prediction_type: null
+pretrained_model_name_or_path: Manojb/stable-diffusion-2-1-base
+push_to_hub: true
+random_flip: true
+rank: 4
+report_to: tensorboard
+resolution: 512
+resume_from_checkpoint: null
+revision: null
+scale_lr: false
+seed: 42
+snr_gamma: null
+train_batch_size: 4
+train_data_dir: null
+use_8bit_adam: false
+validation_epochs: 1
+validation_prompt: a car is driving down the road with another car
+variant: null

logs/text2image-fine-tune/events.out.tfevents.1771485530.wmgubws17.wmgds.wmg.warwick.ac.uk.3161872.0

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

pytorch_lora_weights.safetensors

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