p1atdev
/

plat-diffusion

StableDiffusionPipeline

stable-diffusion

stable-diffusion-diffusers

Model card Files Files and versions

Update README.md

#4

by p1atdev - opened Jan 4, 2023

base: refs/heads/main

←

from: refs/pr/4

Discussion Files changed

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -12,6 +12,8 @@ tags:
 Plat Diffusion is a fine-tuned model based on [Waifu Diffusion v1.4 Anime Epoch 1](https://huggingface.co/hakurei/waifu-diffusion-v1-4) with images generated with niji・journey.
 ![image.png](https://s3.amazonaws.com/moonup/production/uploads/1672836449977-6305db1fcfbde33ef7d480ff.png)
@@ -33,7 +35,7 @@ from diffusers import StableDiffusionPipeline
 import torch
 model_id = "p1atdev/plat-diffusion"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 prompt = "masterpiece, best quality, 1girl, orange feather, blue crystals, blurry foreground"

 Plat Diffusion is a fine-tuned model based on [Waifu Diffusion v1.4 Anime Epoch 1](https://huggingface.co/hakurei/waifu-diffusion-v1-4) with images generated with niji・journey.
+`.safetensors` file is [here](https://huggingface.co/p1atdev/pd-archive/tree/main).
 ![image.png](https://s3.amazonaws.com/moonup/production/uploads/1672836449977-6305db1fcfbde33ef7d480ff.png)
 import torch
 model_id = "p1atdev/plat-diffusion"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32) # not working with float16
 pipe = pipe.to("cuda")
 prompt = "masterpiece, best quality, 1girl, orange feather, blue crystals, blurry foreground"