anonymous commited on
Commit
fb1b680
1 Parent(s): 2a8678c
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -296,6 +296,7 @@ def process1(*args):
296
  detector = global_state.detector
297
  controller = global_state.controller
298
  model.control_scales = [cfg.control_strength] * 13
 
299
 
300
  num_samples = 1
301
  eta = 0.0
 
296
  detector = global_state.detector
297
  controller = global_state.controller
298
  model.control_scales = [cfg.control_strength] * 13
299
+ model.to(device)
300
 
301
  num_samples = 1
302
  eta = 0.0