multimodalart HF staff commited on
Commit
8c21422
1 Parent(s): 4bacbb7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -65,8 +65,8 @@ def infer(prompt, ip_adapter_image, negative_prompt="", seed=42, randomize_seed=
65
  seed = random.randint(0, MAX_SEED)
66
 
67
  generator = torch.Generator(device="cuda").manual_seed(seed)
68
- pipe = pipe.to("cuda")
69
- image_encoder = image_encoder.to("cuda")
70
  pipe.image_encoder = image_encoder
71
  pipe.set_ip_adapter_scale([ip_adapter_scale])
72
 
 
65
  seed = random.randint(0, MAX_SEED)
66
 
67
  generator = torch.Generator(device="cuda").manual_seed(seed)
68
+ pipe.to("cuda")
69
+ image_encoder.to("cuda")
70
  pipe.image_encoder = image_encoder
71
  pipe.set_ip_adapter_scale([ip_adapter_scale])
72