End of training

README.md

@@ -0,0 +1,78 @@
+
+---
+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 - stillerman/trdne-simple-prompt
+These are LoRA adaption weights for runwayml/stable-diffusion-v1-5. The weights were fine-tuned on the stillerman/rugs-1.9k-downloaded 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)
+
+
+
+## Training Args
+```json
+{
+  "pretrained_model_name_or_path": "runwayml/stable-diffusion-v1-5",
+  "revision": null,
+  "dataset_name": "stillerman/rugs-1.9k-downloaded",
+  "dataset_config_name": null,
+  "train_data_dir": null,
+  "image_column": "image",
+  "caption_column": "caption",
+  "validation_prompt": "top down view of an antique rug from Peter Pap's collection",
+  "num_validation_images": 4,
+  "validation_epochs": 1,
+  "max_train_samples": null,
+  "output_dir": "model_out_dir",
+  "cache_dir": null,
+  "seed": 1337,
+  "resolution": 512,
+  "center_crop": true,
+  "random_flip": true,
+  "train_batch_size": 1,
+  "num_train_epochs": 1,
+  "max_train_steps": 250,
+  "gradient_accumulation_steps": 4,
+  "gradient_checkpointing": false,
+  "learning_rate": 0.0001,
+  "scale_lr": false,
+  "lr_scheduler": "cosine",
+  "lr_warmup_steps": 0,
+  "snr_gamma": null,
+  "use_8bit_adam": false,
+  "allow_tf32": false,
+  "dataloader_num_workers": 8,
+  "adam_beta1": 0.9,
+  "adam_beta2": 0.999,
+  "adam_weight_decay": 0.01,
+  "adam_epsilon": 1e-08,
+  "max_grad_norm": 1.0,
+  "push_to_hub": true,
+  "hub_token": null,
+  "prediction_type": null,
+  "hub_model_id": "trdne-simple-prompt",
+  "logging_dir": "logs",
+  "mixed_precision": null,
+  "report_to": "tensorboard",
+  "local_rank": -1,
+  "checkpointing_steps": 250,
+  "checkpoints_total_limit": null,
+  "resume_from_checkpoint": null,
+  "enable_xformers_memory_efficient_attention": false,
+  "noise_offset": 0,
+  "rank": 4,
+  "override_caption": "top down view of an antique rug from Peter Pap's collection"
+}
+```

checkpoint-250/optimizer.bin

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

checkpoint-250/pytorch_model.bin

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

checkpoint-250/random_states_0.pkl

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

checkpoint-250/scaler.pt

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

checkpoint-250/scheduler.bin

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

logs/text2image-fine-tune/1691078329.4588976/events.out.tfevents.1691078329.c3668f57c6d2.3857.1

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

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

@@ -0,0 +1,50 @@
+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: 250
+checkpoints_total_limit: null
+dataloader_num_workers: 8
+dataset_config_name: null
+dataset_name: stillerman/rugs-1.9k-downloaded
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 4
+gradient_checkpointing: false
+hub_model_id: trdne-simple-prompt
+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: 250
+mixed_precision: null
+noise_offset: 0
+num_train_epochs: 1
+num_validation_images: 4
+output_dir: model_out_dir
+override_caption: top down view of an antique rug from Peter Pap's collection
+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: 1337
+snr_gamma: null
+train_batch_size: 1
+train_data_dir: null
+use_8bit_adam: false
+validation_epochs: 1
+validation_prompt: top down view of an antique rug from Peter Pap's collection

logs/text2image-fine-tune/events.out.tfevents.1691078329.c3668f57c6d2.3857.0

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

pytorch_lora_weights.bin

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