Update app.py

app.py

@@ -139,7 +139,7 @@ def compute_image_state(image):
         image = image.to(device)
         image_features = visual_encoder.encode_images(image.unsqueeze(0)).squeeze(0) # [L, D]
         # apply layer norm to image feature, very important
-        image_features = F.layer_norm(image_features, 
+        image_features = F.layer_norm(image_features.to(torch.float32), 
                                     (image_features.shape[-1],), 
                                     weight=model.w['blocks.0.ln0.weight'], 
                                     bias=model.w['blocks.0.ln0.bias'])