animrods commited on
Commit
b248f4e
·
verified ·
1 Parent(s): 842e3ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -25,9 +25,6 @@ def predict(prompt, ip_adapter_image, ip_adapter_scale=0.5, negative_prompt="",
25
  ip_adapter_image.resize((224,224))
26
 
27
  generator = torch.Generator(device="cuda").manual_seed(seed)
28
- pipe.to("cuda")
29
- image_encoder.to("cuda")
30
- pipe.image_encoder = image_encoder
31
  pipe.set_ip_adapter_scale([ip_adapter_scale])
32
 
33
  image = pipe(
 
25
  ip_adapter_image.resize((224,224))
26
 
27
  generator = torch.Generator(device="cuda").manual_seed(seed)
 
 
 
28
  pipe.set_ip_adapter_scale([ip_adapter_scale])
29
 
30
  image = pipe(