alfredplpl
commited on
Commit
•
debc4f2
1
Parent(s):
6a9484b
Update README.md
Browse files
README.md
CHANGED
@@ -92,7 +92,7 @@ import torch
|
|
92 |
model_id = "aipicasso/cool-japan-diffusion-2-1-1"
|
93 |
|
94 |
scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
|
95 |
-
pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
|
96 |
pipe = pipe.to("cuda")
|
97 |
|
98 |
prompt = "anime, masterpiece, a portrait of a girl, good pupil, 4k, detailed"
|
|
|
92 |
model_id = "aipicasso/cool-japan-diffusion-2-1-1"
|
93 |
|
94 |
scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
|
95 |
+
pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
|
96 |
pipe = pipe.to("cuda")
|
97 |
|
98 |
prompt = "anime, masterpiece, a portrait of a girl, good pupil, 4k, detailed"
|