Spaces:
Runtime error
Runtime error
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 ,
|