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 |
|