End of training

README.md

@@ -14,7 +14,7 @@ inference: true
     
 # Text-to-image finetuning - MaxReynolds/MyPatternModel
 
-This pipeline was finetuned from **CompVis/stable-diffusion-v1-4** on the **MaxReynolds/MyPatternDataset** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: [',<r4nd0m-l4b3l>']: 
+This pipeline was finetuned from **CompVis/stable-diffusion-v1-4** on the **MaxReynolds/MyPatternDataset** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: ['<r4nd0m-l4b3l>']: 
 
 ![val_imgs_grid](./val_imgs_grid.png)
 
@@ -28,7 +28,7 @@ from diffusers import DiffusionPipeline
 import torch
 
 pipeline = DiffusionPipeline.from_pretrained("MaxReynolds/MyPatternModel", torch_dtype=torch.float16)
-prompt = ",<r4nd0m-l4b3l>"
+prompt = "<r4nd0m-l4b3l>"
 image = pipeline(prompt).images[0]
 image.save("my_image.png")
 ```
@@ -37,7 +37,7 @@ image.save("my_image.png")
 
 These are the key hyperparameters used during training:
 
-* Epochs: 32
+* Epochs: 22
 * Learning rate: 1e-05
 * Batch size: 1
 * Gradient accumulation steps: 4
@@ -45,4 +45,4 @@ These are the key hyperparameters used during training:
 * Mixed-precision: fp16
 
 
-More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/max-f-reynolds/text2image-fine-tune/runs/8v2yfdkf).
+More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/max-f-reynolds/text2image-fine-tune/runs/kybv4sem).

feature_extractor/preprocessor_config.json

@@ -24,5 +24,6 @@
   "rescale_factor": 0.00392156862745098,
   "size": {
     "shortest_edge": 224
-  }
+  },
+  "use_square_size": false
 }

model_index.json

@@ -1,6 +1,6 @@
 {
   "_class_name": "StableDiffusionPipeline",
-  "_diffusers_version": "0.22.0.dev0",
+  "_diffusers_version": "0.23.0.dev0",
   "_name_or_path": "CompVis/stable-diffusion-v1-4",
   "feature_extractor": [
     "transformers",

safety_checker/config.json

@@ -15,7 +15,7 @@
     "num_attention_heads": 12
   },
   "torch_dtype": "float32",
-  "transformers_version": "4.34.1",
+  "transformers_version": "4.35.0",
   "vision_config": {
     "dropout": 0.0,
     "hidden_size": 1024,

scheduler/scheduler_config.json

@@ -1,6 +1,6 @@
 {
   "_class_name": "PNDMScheduler",
-  "_diffusers_version": "0.22.0.dev0",
+  "_diffusers_version": "0.23.0.dev0",
   "beta_end": 0.012,
   "beta_schedule": "scaled_linear",
   "beta_start": 0.00085,

text_encoder/config.json

@@ -20,6 +20,6 @@
   "pad_token_id": 1,
   "projection_dim": 512,
   "torch_dtype": "float16",
-  "transformers_version": "4.34.1",
+  "transformers_version": "4.35.0",
   "vocab_size": 49408
 }

unet/config.json

@@ -1,6 +1,6 @@
 {
   "_class_name": "UNet2DConditionModel",
-  "_diffusers_version": "0.22.0.dev0",
+  "_diffusers_version": "0.23.0.dev0",
   "_name_or_path": "CompVis/stable-diffusion-v1-4",
   "act_fn": "silu",
   "addition_embed_type": null,

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:371c140ab76858085f1a81995f7e78a11e19deee8d93481373a8a6b7b98f4364
+oid sha256:fc3d75a7ba6037cd8999e81d7be6b02cc0955b0221f2e7189a29cb553b1d7439
 size 3438167536

vae/config.json

@@ -1,6 +1,6 @@
 {
   "_class_name": "AutoencoderKL",
-  "_diffusers_version": "0.22.0.dev0",
+  "_diffusers_version": "0.23.0.dev0",
   "_name_or_path": "CompVis/stable-diffusion-v1-4",
   "act_fn": "silu",
   "block_out_channels": [