shariqfarooq commited on
Commit
a04dfaa
1 Parent(s): edd75e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.float16)
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.