lemonaddie
commited on
Commit
•
59715a4
1
Parent(s):
2d3d463
Update app.py
Browse files
app.py
CHANGED
@@ -359,7 +359,7 @@ def run_demo_server(pipe):
|
|
359 |
map_id_to_default = {b._id: b.value for b in blocks_settings}
|
360 |
|
361 |
inputs = [
|
362 |
-
input_image,
|
363 |
ensemble_size,
|
364 |
denoise_steps,
|
365 |
processing_res,
|
|
|
359 |
map_id_to_default = {b._id: b.value for b in blocks_settings}
|
360 |
|
361 |
inputs = [
|
362 |
+
input_image.cuda(),
|
363 |
ensemble_size,
|
364 |
denoise_steps,
|
365 |
processing_res,
|