multimodalart HF Staff commited on
Commit
614a9ba
·
verified ·
1 Parent(s): 818f84a

Cast num_samples to int in handler to avoid TypeError on prompt_embeds.repeat

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -180,6 +180,8 @@ def create_image(content_image_pil,
180
  ):
181
 
182
 
 
 
183
  if content_image_pil is None:
184
  content_image_pil = Image.fromarray(
185
  np.zeros((1024, 1024, 3), dtype=np.uint8)).convert('RGB')
 
180
  ):
181
 
182
 
183
+ num_samples = int(num_samples)
184
+
185
  if content_image_pil is None:
186
  content_image_pil = Image.fromarray(
187
  np.zeros((1024, 1024, 3), dtype=np.uint8)).convert('RGB')