Spaces:
Runtime error
Runtime error
TiankaiHang
commited on
Commit
•
106e80b
1
Parent(s):
411b6f0
sync
Browse files
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)
|