TiankaiHang commited on
Commit
106e80b
1 Parent(s): 411b6f0
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -291,6 +291,7 @@ def main(ckpt="checkpoints/v1-5-pruned-emaonly-adaption-task-humanalign.ckpt", a
291
  model.eval().cuda()
292
  else:
293
  model.eval()
 
294
 
295
  model_wrap = K.external.CompVisDenoiser(model)
296
  model_wrap_cfg = CFGDenoiser(model_wrap)
 
291
  model.eval().cuda()
292
  else:
293
  model.eval()
294
+ model = model.float()
295
 
296
  model_wrap = K.external.CompVisDenoiser(model)
297
  model_wrap_cfg = CFGDenoiser(model_wrap)