Fix syntax error and hardcoded device
Browse filesIn code snippet, fix syntax error in positive_magic dictionary due to missing comma, and hardcoded device on pipeline init
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")
|