merve HF Staff commited on
Commit
7502a3f
·
1 Parent(s): 4f4be5c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
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
- p_params = replicate(params)
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=p_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,