Fix syntax error and hardcoded device
#64
by tossowski - opened
README.md
CHANGED
|
@@ -57,7 +57,7 @@ pipe = DiffusionPipeline.from_pretrained(model_name, torch_dtype=torch_dtype)
|
|
| 57 |
pipe = pipe.to(device)
|
| 58 |
|
| 59 |
positive_magic = {
|
| 60 |
-
"en": "Ultra HD, 4K, cinematic composition." # for english prompt,
|
| 61 |
"zh": "超清，4K，电影级构图" # for chinese prompt,
|
| 62 |
}
|
| 63 |
|
|
@@ -87,7 +87,7 @@ image = pipe(
|
|
| 87 |
height=height,
|
| 88 |
num_inference_steps=50,
|
| 89 |
true_cfg_scale=4.0,
|
| 90 |
-
generator=torch.Generator(device=
|
| 91 |
).images[0]
|
| 92 |
|
| 93 |
image.save("example.png")
|
|
|
|
| 57 |
pipe = pipe.to(device)
|
| 58 |
|
| 59 |
positive_magic = {
|
| 60 |
+
"en": "Ultra HD, 4K, cinematic composition.", # for english prompt,
|
| 61 |
"zh": "超清，4K，电影级构图" # for chinese prompt,
|
| 62 |
}
|
| 63 |
|
|
|
|
| 87 |
height=height,
|
| 88 |
num_inference_steps=50,
|
| 89 |
true_cfg_scale=4.0,
|
| 90 |
+
generator=torch.Generator(device=device).manual_seed(42)
|
| 91 |
).images[0]
|
| 92 |
|
| 93 |
image.save("example.png")
|