Spaces:
Runtime error
Runtime error
Commit
·
a04dfaa
1
Parent(s):
edd75e6
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ from loosecontrol import LooseControlNet
|
|
| 12 |
|
| 13 |
device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
|
| 14 |
cn = LooseControlNet()
|
| 15 |
-
cn.pipe = cn.pipe.to(torch_device=device, torch_dtype=torch.
|
| 16 |
|
| 17 |
# Need to figure out a better way how to do this per user, making 'cf attention' act like a state per user.
|
| 18 |
# For now, we just copy the model.
|
|
|
|
| 12 |
|
| 13 |
device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
|
| 14 |
cn = LooseControlNet()
|
| 15 |
+
cn.pipe = cn.pipe.to(torch_device=device, torch_dtype=torch.float32)
|
| 16 |
|
| 17 |
# Need to figure out a better way how to do this per user, making 'cf attention' act like a state per user.
|
| 18 |
# For now, we just copy the model.
|