animrods commited on
Commit
c8d7ddc
·
verified ·
1 Parent(s): 0644651

Update app.py

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