Spaces:

pollen-robotics
/

reachy_mini_conversation_app

Running

Alina Lozovskaya commited on Sep 4

Commit

14afe2b

1 Parent(s): c5b1e86

Replace torch_dtype to dtype

Files changed (1) hide show

src/reachy_mini_conversation_demo/vision.py CHANGED Viewed

@@ -70,7 +70,7 @@ class VisionProcessor:
             else:
                 dtype = torch.float32
-            model_kwargs = {"torch_dtype": dtype}
             # flash_attention_2 is CUDA-only; skip on MPS/CPU
             if self.device == "cuda":

             else:
                 dtype = torch.float32
+            model_kwargs = {"dtype": dtype}
             # flash_attention_2 is CUDA-only; skip on MPS/CPU
             if self.device == "cuda":