Spaces:
Running
on
L40S
Running
on
L40S
Commit
•
3c1793b
1
Parent(s):
d4c4f8f
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,7 +32,7 @@ pipe = StableDiffusion3Pipeline.from_pretrained(
|
|
| 32 |
model_path,
|
| 33 |
transformer=transformer,
|
| 34 |
torch_dtype=torch.bfloat16
|
| 35 |
-
)
|
| 36 |
|
| 37 |
pipe.init_ipadapter(
|
| 38 |
ip_adapter_path=ipadapter_path,
|
|
|
|
| 32 |
model_path,
|
| 33 |
transformer=transformer,
|
| 34 |
torch_dtype=torch.bfloat16
|
| 35 |
+
).to("cuda")
|
| 36 |
|
| 37 |
pipe.init_ipadapter(
|
| 38 |
ip_adapter_path=ipadapter_path,
|