Update README.md
Browse files
README.md
CHANGED
|
@@ -18,7 +18,9 @@ You can also export the model to [ONNX](https://huggingface.co/docs/diffusers/op
|
|
| 18 |
from diffusers import StableDiffusionPipeline
|
| 19 |
import torch
|
| 20 |
model_id = "michaelz/arcaneface-diffusion"
|
| 21 |
-
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.
|
|
|
|
|
|
|
| 22 |
pipe = pipe.to("cuda")
|
| 23 |
prompt = "arcaneface, a beautiful girl"
|
| 24 |
image = pipe(prompt).images[0]
|
|
|
|
| 18 |
from diffusers import StableDiffusionPipeline
|
| 19 |
import torch
|
| 20 |
model_id = "michaelz/arcaneface-diffusion"
|
| 21 |
+
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
|
| 22 |
+
pipe.requires_safety_checker = False
|
| 23 |
+
pipe.safety_checker = None
|
| 24 |
pipe = pipe.to("cuda")
|
| 25 |
prompt = "arcaneface, a beautiful girl"
|
| 26 |
image = pipe(prompt).images[0]
|