Spaces:
Runtime error
Runtime error
anonymous
commited on
Commit
·
fb1b680
1
Parent(s):
2a8678c
update
Browse files
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
|