End of training

README.md

@@ -1,7 +1,7 @@
 
 ---
 license: creativeml-openrail-m
-base_model: CompVis/stable-diffusion-v1-2
+base_model: MaxReynolds/SouderRocketLauncherNetCombined-SD1-5
 datasets:
 - MaxReynolds/Lee_Souder_RocketLauncher_Generated
 tags:
@@ -14,7 +14,7 @@ inference: true
     
 # Text-to-image finetuning - MaxReynolds/SouderRocketLauncherNetCombinedGenerated
 
-This pipeline was finetuned from **CompVis/stable-diffusion-v1-2** on the **MaxReynolds/Lee_Souder_RocketLauncher_Generated** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: ['Rocket Launcher by Lee Souder']: 
+This pipeline was finetuned from **MaxReynolds/SouderRocketLauncherNetCombined-SD1-5** on the **MaxReynolds/Lee_Souder_RocketLauncher_Generated** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: ['Rocket Launcher by Lee Souder']: 
 
 ![val_imgs_grid](./val_imgs_grid.png)
 
@@ -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/6lr38hd2).
+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/18quem9n).

model_index.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "StableDiffusionPipeline",
   "_diffusers_version": "0.22.0.dev0",
-  "_name_or_path": "CompVis/stable-diffusion-v1-2",
+  "_name_or_path": "MaxReynolds/SouderRocketLauncherNetCombined-SD1-5",
   "feature_extractor": [
     "transformers",
     "CLIPImageProcessor"

safety_checker/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "/root/.cache/huggingface/hub/models--CompVis--stable-diffusion-v1-2/snapshots/f519eebf1372598b85ab317ec273a63ebeddd5eb/safety_checker",
+  "_name_or_path": "/root/.cache/huggingface/hub/models--MaxReynolds--SouderRocketLauncherNetCombined-SD1-5/snapshots/b69fc3044d90f984b55b42d72304e0b9817aec63/safety_checker",
   "architectures": [
     "StableDiffusionSafetyChecker"
   ],

scheduler/scheduler_config.json

@@ -4,6 +4,7 @@
   "beta_end": 0.012,
   "beta_schedule": "scaled_linear",
   "beta_start": 0.00085,
+  "clip_sample": false,
   "num_train_timesteps": 1000,
   "prediction_type": "epsilon",
   "set_alpha_to_one": false,

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "CompVis/stable-diffusion-v1-2",
+  "_name_or_path": "MaxReynolds/SouderRocketLauncherNetCombined-SD1-5",
   "architectures": [
     "CLIPTextModel"
   ],
@@ -18,7 +18,7 @@
   "num_attention_heads": 12,
   "num_hidden_layers": 12,
   "pad_token_id": 1,
-  "projection_dim": 512,
+  "projection_dim": 768,
   "torch_dtype": "float16",
   "transformers_version": "4.34.1",
   "vocab_size": 49408

tokenizer/special_tokens_map.json

@@ -13,7 +13,13 @@
     "rstrip": false,
     "single_word": false
   },
-  "pad_token": "<|endoftext|>",
+  "pad_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
   "unk_token": {
     "content": "<|endoftext|>",
     "lstrip": false,

tokenizer/tokenizer_config.json

@@ -18,6 +18,7 @@
       "special": true
     }
   },
+  "additional_special_tokens": [],
   "bos_token": "<|startoftext|>",
   "clean_up_tokenization_spaces": true,
   "do_lower_case": true,

unet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.22.0.dev0",
-  "_name_or_path": "CompVis/stable-diffusion-v1-2",
+  "_name_or_path": "MaxReynolds/SouderRocketLauncherNetCombined-SD1-5",
   "act_fn": "silu",
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:94e0a36a276cedf293ee81a0b592f93d9852e9849dc617fa4a4addc815137076
+oid sha256:20c8301083108224f50ede97ffa76b514d90bb3cf2fd671e6f0330b6c196fd59
 size 3438167536

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.22.0.dev0",
-  "_name_or_path": "CompVis/stable-diffusion-v1-2",
+  "_name_or_path": "MaxReynolds/SouderRocketLauncherNetCombined-SD1-5",
   "act_fn": "silu",
   "block_out_channels": [
     128,