Spaces:
Paused
Paused
shariqfarooq
commited on
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.
|