Update app.py
Browse files
app.py
CHANGED
|
@@ -56,15 +56,12 @@ def infer(prompts, negative_prompts, image):
|
|
| 56 |
negative_prompt_ids = pipe.prepare_text_inputs([negative_prompts] * num_samples)
|
| 57 |
processed_image = pipe.prepare_image_inputs([canny_image] * num_samples)
|
| 58 |
|
| 59 |
-
|
| 60 |
-
prompt_ids = shard(prompt_ids)
|
| 61 |
-
negative_prompt_ids = shard(negative_prompt_ids)
|
| 62 |
-
processed_image = shard(processed_image)
|
| 63 |
|
| 64 |
output = pipe(
|
| 65 |
prompt_ids=prompt_ids,
|
| 66 |
image=processed_image,
|
| 67 |
-
params=
|
| 68 |
prng_seed=rng,
|
| 69 |
num_inference_steps=50,
|
| 70 |
neg_prompt_ids=negative_prompt_ids,
|
|
|
|
| 56 |
negative_prompt_ids = pipe.prepare_text_inputs([negative_prompts] * num_samples)
|
| 57 |
processed_image = pipe.prepare_image_inputs([canny_image] * num_samples)
|
| 58 |
|
| 59 |
+
|
|
|
|
|
|
|
|
|
|
| 60 |
|
| 61 |
output = pipe(
|
| 62 |
prompt_ids=prompt_ids,
|
| 63 |
image=processed_image,
|
| 64 |
+
params=params,
|
| 65 |
prng_seed=rng,
|
| 66 |
num_inference_steps=50,
|
| 67 |
neg_prompt_ids=negative_prompt_ids,
|