sync

app.py

@@ -291,7 +291,6 @@ def main(ckpt="checkpoints/v1-5-pruned-emaonly-adaption-task-humanalign.ckpt", a
         model.eval().cuda()
     else:
         model.eval()
-        model = model.float()
 
     model_wrap = K.external.CompVisDenoiser(model)
     model_wrap_cfg = CFGDenoiser(model_wrap)

stable_diffusion/ldm/modules/attention.py

@@ -379,7 +379,8 @@ class SpatialTransformer(nn.Module):
         x_in = x
         if self.force_type_convert:
             x = self.norm.float()(x.float())
-            x = x.half()
+            if torch.cuda.is_available():
+                x = x.half()
         else:
             x = self.norm(x)
         if not self.use_linear: