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,
|