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 |
|