animrods commited on
Commit
8676aa4
·
verified ·
1 Parent(s): 5abbfcd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -24,7 +24,9 @@ def predict(prompt, ip_adapter_image, ip_adapter_scale=0.5, negative_prompt="",
24
  seed = random.randint(0, MAX_SEED)
25
 
26
  if not center_crop:
27
- ip_adapter_image = [x.resize((224,224)) for x in ip_adapter_image]
 
 
28
 
29
  generator = torch.Generator(device="cuda").manual_seed(seed)
30
  pipe.set_ip_adapter_scale([ip_adapter_scale])
 
24
  seed = random.randint(0, MAX_SEED)
25
 
26
  if not center_crop:
27
+ ip_adapter_image = [x.resize((224,224)) for x in ip_adapter_image[0]]
28
+ else:
29
+ ip_adapter_image = ip_adapter_image[0]
30
 
31
  generator = torch.Generator(device="cuda").manual_seed(seed)
32
  pipe.set_ip_adapter_scale([ip_adapter_scale])