lemonaddie commited on
Commit
b82911a
1 Parent(s): bf42f00

Update app2.py

Browse files
Files changed (1) hide show
  1. app2.py +1 -0
app2.py CHANGED
@@ -78,6 +78,7 @@ except:
78
  pass # run without xformers
79
 
80
  pipe = pipe.to(device)
 
81
  #run_demo_server(pipe)
82
 
83
  @spaces.GPU
 
78
  pass # run without xformers
79
 
80
  pipe = pipe.to(device)
81
+ pipe_cfg = pipe_cfg.to(device)
82
  #run_demo_server(pipe)
83
 
84
  @spaces.GPU