Spaces:
Running
on
Zero
Running
on
Zero
lixiang46
commited on
Commit
•
595a73a
1
Parent(s):
3936987
fix bug and change seed
Browse files
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 =
|
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 =
|
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 ,
|