Update README.md
Browse files
README.md
CHANGED
@@ -12,6 +12,8 @@ tags:
|
|
12 |
|
13 |
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.
|
14 |
|
|
|
|
|
15 |
![image.png](https://s3.amazonaws.com/moonup/production/uploads/1672836449977-6305db1fcfbde33ef7d480ff.png)
|
16 |
|
17 |
|
@@ -33,7 +35,7 @@ from diffusers import StableDiffusionPipeline
|
|
33 |
import torch
|
34 |
|
35 |
model_id = "p1atdev/plat-diffusion"
|
36 |
-
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16
|
37 |
pipe = pipe.to("cuda")
|
38 |
|
39 |
prompt = "masterpiece, best quality, 1girl, orange feather, blue crystals, blurry foreground"
|
|
|
12 |
|
13 |
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.
|
14 |
|
15 |
+
`.safetensors` file is [here](https://huggingface.co/p1atdev/pd-archive/tree/main).
|
16 |
+
|
17 |
![image.png](https://s3.amazonaws.com/moonup/production/uploads/1672836449977-6305db1fcfbde33ef7d480ff.png)
|
18 |
|
19 |
|
|
|
35 |
import torch
|
36 |
|
37 |
model_id = "p1atdev/plat-diffusion"
|
38 |
+
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32) # not working with float16
|
39 |
pipe = pipe.to("cuda")
|
40 |
|
41 |
prompt = "masterpiece, best quality, 1girl, orange feather, blue crystals, blurry foreground"
|