lixiang46 commited on
Commit
595a73a
1 Parent(s): 3936987

fix bug and change seed

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -166,7 +166,7 @@ def infer_canny(prompt,
166
  def infer_pose(prompt,
167
  image = None,
168
  negative_prompt = "nsfw,脸部阴影,低分辨率,jpeg伪影、模糊、糟糕,黑脸,霓虹灯",
169
- seed = 397886929,
170
  randomize_seed = False,
171
  guidance_scale = 6.0,
172
  num_inference_steps = 50,
@@ -178,7 +178,7 @@ def infer_pose(prompt,
178
  seed = random.randint(0, MAX_SEED)
179
  generator = torch.Generator().manual_seed(seed)
180
  init_image = resize_image(image, MAX_IMAGE_SIZE)
181
- pipe = pipe_canny.to("cuda")
182
  condi_img = process_dwpose_condition(np.array(init_image), MAX_IMAGE_SIZE)
183
  image = pipe(
184
  prompt= prompt ,
 
166
  def infer_pose(prompt,
167
  image = None,
168
  negative_prompt = "nsfw,脸部阴影,低分辨率,jpeg伪影、模糊、糟糕,黑脸,霓虹灯",
169
+ seed = 66,
170
  randomize_seed = False,
171
  guidance_scale = 6.0,
172
  num_inference_steps = 50,
 
178
  seed = random.randint(0, MAX_SEED)
179
  generator = torch.Generator().manual_seed(seed)
180
  init_image = resize_image(image, MAX_IMAGE_SIZE)
181
+ pipe = pipe_pose.to("cuda")
182
  condi_img = process_dwpose_condition(np.array(init_image), MAX_IMAGE_SIZE)
183
  image = pipe(
184
  prompt= prompt ,