Update app.py

app.py

@@ -381,7 +381,13 @@ def load_model(source: str):
             nn.init.zeros_(model.proj.bias)
     else:
         model.load_state_dict(state, strict=True)
-
+    
+    # 🔧 ensure everything is float32 (fixes mixed-dtype LayerNorm on CPU)
+    model = model.to(torch.float32)
+    for bname, buf in model.named_buffers():
+        if buf.dtype.is_floating_point:
+            setattr(model, bname, buf.float())
+    
     model.eval()
     return model, tokenizer, conf["radius"]