tossowski commited on
Commit
e4ce9ed
·
verified ·
1 Parent(s): 9e31a1d

Fix syntax error and hardcoded device

Browse files

In code snippet, fix syntax error in positive_magic dictionary due to missing comma, and hardcoded device on pipeline init

Files changed (1) hide show
  1. README.md +2 -2
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="cuda").manual_seed(42)
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")