Update app.py

app.py

@@ -135,7 +135,7 @@ def generate(in_im_embs):
     im_emb, _ = pipe.encode_image(
                 output.frames[0][len(output.frames[0])//2], DEVICE, 1, output_hidden_state
             )
-    im_emb = im_emb.detach().to('cpu')
+    im_emb = im_emb.detach().to('cpu').to(torch.float32)
 
     nsfw = maybe_nsfw(output.frames[0][len(output.frames[0])//2])
     
@@ -467,7 +467,7 @@ def encode_space(x):
     im_emb, _ = pipe.encode_image(
                 image, DEVICE, 1, output_hidden_state
             )
-    return im_emb.detach().to('cpu')
+    return im_emb.detach().to('cpu').to(torch.float32)
 
 # prep our calibration prompts
 for im in [