Update README.md
Browse files
README.md
CHANGED
@@ -20,14 +20,13 @@ You can also export the model to [ONNX](https://huggingface.co/docs/diffusers/op
|
|
20 |
from diffusers import StableDiffusionPipeline
|
21 |
import torch
|
22 |
|
23 |
-
model_id = "
|
24 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
25 |
pipe = pipe.to("cuda")
|
26 |
|
27 |
prompt = "a beautiful young female with long dark hair, clothed in full dress"
|
28 |
image = pipe(prompt).images[0]
|
29 |
-
|
30 |
-
image.save("./samples/0_0.png")
|
31 |
```
|
32 |
|
33 |
|
|
|
20 |
from diffusers import StableDiffusionPipeline
|
21 |
import torch
|
22 |
|
23 |
+
model_id = "AlexZheng/galactic-diffusion-v1.0"
|
24 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
25 |
pipe = pipe.to("cuda")
|
26 |
|
27 |
prompt = "a beautiful young female with long dark hair, clothed in full dress"
|
28 |
image = pipe(prompt).images[0]
|
29 |
+
image.save("./samples/1.png")
|
|
|
30 |
```
|
31 |
|
32 |
|