lemonaddie commited on
Commit
59715a4
1 Parent(s): 2d3d463

Update app.py

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