Error when training model on M3 Max MPS

Help scvi-tools
6

Please refer to the following thread: Macbook M1 M2 mps acceleration with scVI - #4 by martinkim0