mokady commited on
Commit
6e51693
1 Parent(s): d81f726

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def predict(prompt, ip_adapter_image, ip_adapter_scale=0.5, negative_prompt="",
26
  if not center_crop:
27
  ip_adapter_image = ip_adapter_image.resize((224,224))
28
 
29
- generator = torch.Generator(device="cpu").manual_seed(seed)
30
  pipe.set_ip_adapter_scale([ip_adapter_scale])
31
 
32
  image = pipe(
 
26
  if not center_crop:
27
  ip_adapter_image = ip_adapter_image.resize((224,224))
28
 
29
+ generator = torch.Generator(device="cuda").manual_seed(seed)
30
  pipe.set_ip_adapter_scale([ip_adapter_scale])
31
 
32
  image = pipe(