def test_sharded_checkpoints_device_map(self):
        """
        tests/models/autoencoders/test_models_autoencoder_mochi.py::AutoencoderKLMochiTests::test_sharded_checkpoints_device_map -
        RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:5!
        """

        tests/models/autoencoders/test_models_autoencoder_mochi.py::AutoencoderKLMochiTests::test_sharded_checkpoints_device_map -
        RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:5!
        
tests/models/autoencoders/test_models_autoencoder_mochi.py