ValueError: wo was specified in the `_keep_in_fp32_modules` list, but is not part of the modules in HuskyVisionModel

I get the following error when trying to run demo/inference.py

```bash
ValueError: wo was specified in the `_keep_in_fp32_modules` list, but is not part of the modules in HuskyVisionModel
```

Solved it by commenting line 392 in modeling_husky_embody2.py

 ```python
 # _keep_in_fp32_modules = ["wo"]
```
