Commit
•
7f31109
1
Parent(s):
9d21a93
Update app.py
Browse files
app.py
CHANGED
|
@@ -65,7 +65,7 @@ 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 |
-
|
| 69 |
pipe.set_ip_adapter_scale([ip_adapter_scale])
|
| 70 |
|
| 71 |
image = pipe(
|
|
|
|
| 65 |
seed = random.randint(0, MAX_SEED)
|
| 66 |
|
| 67 |
generator = torch.Generator(device="cuda").manual_seed(seed)
|
| 68 |
+
pipe = pipe.to("cuda")
|
| 69 |
pipe.set_ip_adapter_scale([ip_adapter_scale])
|
| 70 |
|
| 71 |
image = pipe(
|