Add WD 1.3 float16 weights

README.md

@@ -13,6 +13,7 @@ inference: false
 
 waifu-diffusion is a latent text-to-image diffusion model that has been conditioned on high-quality anime images through fine-tuning.
 
+<<<<<<< HEAD
 <img src=https://i.imgur.com/Y5Tmw1S.png width=75% height=75%>
 
 [Original Weights](https://huggingface.co/hakurei/waifu-diffusion-v1-3)
@@ -22,11 +23,15 @@ waifu-diffusion is a latent text-to-image diffusion model that has been conditio
 We also support a [Gradio](https://github.com/gradio-app/gradio) Web UI and Colab with Diffusers to run Waifu Diffusion:
 [![Open In Spaces](https://camo.githubusercontent.com/00380c35e60d6b04be65d3d94a58332be5cc93779f630bcdfc18ab9a3a7d3388/68747470733a2f2f696d672e736869656c64732e696f2f62616467652f25463025394625413425393725323048756767696e67253230466163652d5370616365732d626c7565)](https://huggingface.co/spaces/hakurei/waifu-diffusion-demo)
 [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1_8wPN7dJO746QXsFnB09Uq2VGgSRFuYE#scrollTo=1HaCauSq546O)
+=======
+<img src=https://cdn.discordapp.com/attachments/930499731451428926/1017258164439220254/unknown.png width=20% height=20%>
+>>>>>>> b45bafccd9d0e0757b70a54c7ebc32ff56ca9ee1
 
 ## Model Description
 
 [See here for a full model overview.](https://gist.github.com/harubaru/f727cedacae336d1f7877c4bbe2196e1)
 
+<<<<<<< HEAD
 ## License
 
 This model is open access and available to all, with a CreativeML OpenRAIL-M license further specifying rights and usage.
@@ -36,6 +41,15 @@ The CreativeML OpenRAIL License specifies:
 2. The authors claims no rights on the outputs you generate, you are free to use them and are accountable for their use which must not go against the provisions set in the license
 3. You may re-distribute the weights and use the model commercially and/or as a service. If you do, please be aware you have to include the same use restrictions as the ones in the license and share a copy of the CreativeML OpenRAIL-M to all your users (please read the license entirely and carefully)
 [Please read the full license here](https://huggingface.co/spaces/CompVis/stable-diffusion-license)
+=======
+The current model has been fine-tuned with a learning rate of 5.0e-6 for 4 epochs on 56k Danbooru text-image pairs which all have an aesthetic rating greater than `6.0`.
+
+## Training Data & Annotative Prompting
+
+The data used for fine-tuning has come from a random sample of 56k Danbooru images, which were filtered based on [CLIP Aesthetic Scoring](https://github.com/christophschuhmann/improved-aesthetic-predictor) where only images with an aesthetic score greater than `6.0` were used.
+
+Captions are Danbooru-style captions.
+>>>>>>> b45bafccd9d0e0757b70a54c7ebc32ff56ca9ee1
 
 ## Downstream Uses
 
@@ -53,7 +67,15 @@ pipe = StableDiffusionPipeline.from_pretrained(
     torch_dtype=torch.float32
 ).to('cuda')
 
+<<<<<<< HEAD
 prompt = "1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt"
+=======
+
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision='fp16')
+pipe = pipe.to(device)
+
+prompt = "touhou hakurei_reimu 1girl solo portrait"
+>>>>>>> b45bafccd9d0e0757b70a54c7ebc32ff56ca9ee1
 with autocast("cuda"):
     image = pipe(prompt, guidance_scale=6)["sample"][0]  
     

model_index.json

@@ -11,7 +11,11 @@
   ],
   "scheduler": [
     "diffusers",
+<<<<<<< HEAD
     "LMSDiscreteScheduler"
+=======
+    "DDIMScheduler"
+>>>>>>> b45bafccd9d0e0757b70a54c7ebc32ff56ca9ee1
   ],
   "text_encoder": [
     "transformers",

safety_checker/config.json

@@ -1,6 +1,10 @@
 {
+<<<<<<< HEAD
   "_commit_hash": null,
   "_name_or_path": "CompVis/stable-diffusion-safety-checker",
+=======
+  "_name_or_path": "waifu-diffusion/safety_checker",
+>>>>>>> b45bafccd9d0e0757b70a54c7ebc32ff56ca9ee1
   "architectures": [
     "StableDiffusionSafetyChecker"
   ],
@@ -77,7 +81,11 @@
     "top_p": 1.0,
     "torch_dtype": null,
     "torchscript": false,
+<<<<<<< HEAD
     "transformers_version": "4.22.2",
+=======
+    "transformers_version": "4.21.3",
+>>>>>>> b45bafccd9d0e0757b70a54c7ebc32ff56ca9ee1
     "typical_p": 1.0,
     "use_bfloat16": false,
     "vocab_size": 49408
@@ -161,7 +169,11 @@
     "top_p": 1.0,
     "torch_dtype": null,
     "torchscript": false,
+<<<<<<< HEAD
     "transformers_version": "4.22.2",
+=======
+    "transformers_version": "4.21.3",
+>>>>>>> b45bafccd9d0e0757b70a54c7ebc32ff56ca9ee1
     "typical_p": 1.0,
     "use_bfloat16": false
   },

safety_checker/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1d37ca6e57ace94e4c2f03ed0f67b6dc83e1ef1160892074917aa68b28e2afc1
-size 608098599
+oid sha256:6b1dc15150c06764bb60249c8a68b3e31319c66293d800c252b4d400e3e7ea17
+size 295

scheduler/scheduler_config.json

@@ -4,6 +4,13 @@
   "beta_end": 0.012,
   "beta_schedule": "scaled_linear",
   "beta_start": 0.00085,
+<<<<<<< HEAD
   "num_train_timesteps": 1000,
+=======
+  "clip_sample": false,
+  "num_train_timesteps": 1000,
+  "set_alpha_to_one": false,
+  "timestep_values": null,
+>>>>>>> b45bafccd9d0e0757b70a54c7ebc32ff56ca9ee1
   "trained_betas": null
 }

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "openai/clip-vit-large-patch14",
+  "_name_or_path": "waifu-diffusion/text_encoder",
   "architectures": [
     "CLIPTextModel"
   ],
@@ -18,8 +18,13 @@
   "num_attention_heads": 12,
   "num_hidden_layers": 12,
   "pad_token_id": 1,
+<<<<<<< HEAD
   "projection_dim": 768,
   "torch_dtype": "float16",
   "transformers_version": "4.22.2",
+=======
+  "torch_dtype": "float16",
+  "transformers_version": "4.21.3",
+>>>>>>> b45bafccd9d0e0757b70a54c7ebc32ff56ca9ee1
   "vocab_size": 49408
 }

text_encoder/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:88bd85efb0f84e70521633f578715afb2873db4f2615fdfb1f66e99934715865
-size 246184375
+oid sha256:439fc72b1855e991f9c6525d746fa8a7590e2763a9fc7ac077cea4b2e4a1ce93
+size 295

tokenizer/tokenizer_config.json

@@ -19,7 +19,7 @@
   },
   "errors": "replace",
   "model_max_length": 77,
-  "name_or_path": "openai/clip-vit-large-patch14",
+  "name_or_path": "waifu-diffusion/tokenizer",
   "pad_token": "<|endoftext|>",
   "special_tokens_map_file": "./special_tokens_map.json",
   "tokenizer_class": "CLIPTokenizer",

unet/config.json

@@ -1,6 +1,11 @@
 {
   "_class_name": "UNet2DConditionModel",
+<<<<<<< HEAD
   "_diffusers_version": "0.4.1",
+=======
+  "_diffusers_version": "0.2.4",
+  "_name_or_path": "waifu-diffusion/unet",
+>>>>>>> b45bafccd9d0e0757b70a54c7ebc32ff56ca9ee1
   "act_fn": "silu",
   "attention_head_dim": 8,
   "block_out_channels": [

unet/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:74e8495acd79b59493a5e3512d93d418f9188fb99dd66bd36560e6f7155a82c6
-size 1719312805
+oid sha256:8a9d5548268e0013c7d21a0f451ba016098a7069bbfba4b6744312b58fe86707
+size 297

vae/config.json

@@ -1,6 +1,11 @@
 {
   "_class_name": "AutoencoderKL",
+<<<<<<< HEAD
   "_diffusers_version": "0.4.1",
+=======
+  "_diffusers_version": "0.2.4",
+  "_name_or_path": "waifu-diffusion/vae",
+>>>>>>> b45bafccd9d0e0757b70a54c7ebc32ff56ca9ee1
   "act_fn": "silu",
   "block_out_channels": [
     128,

vae/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:51c8904bc921e1e6f354b5fa8e99a1c82ead2f0540114de21557b8abfbb24ad0
-size 167399505
+oid sha256:9980e77e11dfc7e18e14aefc2604f2bba3db7737e722cd11610f0cf6a1e96271
+size 295