End of training

README.md

@@ -0,0 +1,46 @@
+
+---
+license: creativeml-openrail-m
+base_model: runwayml/stable-diffusion-v1-5
+datasets:
+- lambdalabs/pokemon-blip-captions
+tags:
+- stable-diffusion
+- stable-diffusion-diffusers
+- text-to-image
+- diffusers
+inference: true
+---
+    
+# Text-to-image finetuning - shrikant11/text_to_img_1
+
+This pipeline was finetuned from **runwayml/stable-diffusion-v1-5** on the **lambdalabs/pokemon-blip-captions** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: ['How are you']: 
+
+![val_imgs_grid](./val_imgs_grid.png)
+
+
+## Pipeline usage
+
+You can use the pipeline like so:
+
+```python
+from diffusers import DiffusionPipeline
+import torch
+
+pipeline = DiffusionPipeline.from_pretrained("shrikant11/text_to_img_1", torch_dtype=torch.float16)
+prompt = "How are you"
+image = pipeline(prompt).images[0]
+image.save("my_image.png")
+```
+
+## Training info
+
+These are the key hyperparameters used during training:
+
+* Epochs: 1
+* Learning rate: 1e-05
+* Batch size: 1
+* Gradient accumulation steps: 1
+* Image resolution: 512
+* Mixed-precision: None
+

logs/text2image-fine-tune/1704267462.9875765/events.out.tfevents.1704267462.8792eaab1ee6.29101.1

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

logs/text2image-fine-tune/1704267462.9893029/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: 5
+checkpoints_total_limit: null
+dataloader_num_workers: 0
+dataset_config_name: null
+dataset_name: lambdalabs/pokemon-blip-captions
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 1
+gradient_checkpointing: false
+hub_model_id: null
+hub_token: null
+image_column: image
+input_perturbation: 0
+learning_rate: 1.0e-05
+local_rank: -1
+logging_dir: logs
+lr_scheduler: constant
+lr_warmup_steps: 500
+max_grad_norm: 1.0
+max_train_samples: null
+max_train_steps: 6
+mixed_precision: null
+noise_offset: 0
+non_ema_revision: null
+num_train_epochs: 1
+output_dir: sd-pokemon-model
+prediction_type: null
+pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
+push_to_hub: true
+random_flip: true
+report_to: tensorboard
+resolution: 512
+resume_from_checkpoint: null
+revision: null
+scale_lr: false
+seed: null
+snr_gamma: null
+tracker_project_name: text2image-fine-tune
+train_batch_size: 1
+train_data_dir: null
+use_8bit_adam: false
+use_ema: false
+validation_epochs: 5
+variant: null

logs/text2image-fine-tune/events.out.tfevents.1704267462.8792eaab1ee6.29101.0

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

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3b5d225aeb46ebed774f175585b59d504d7c665e3c6edc67128e4053e04533f3
+oid sha256:030ef58c2770761741d93d7b38cf78ed655f37916882aca6997f2d46a78e74c6
 size 3438167536