Spaces:
Running
on
A10G
Running
on
A10G
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
|