Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
8c21422
1
Parent(s):
4bacbb7
Update app.py
Browse files
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
|
69 |
-
image_encoder
|
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 |
|