p1atdev commited on
Commit
fa7ab76
1 Parent(s): 0cc0f3d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
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"