pass device to swittihf init

models/pipeline.py

@@ -28,7 +28,7 @@ class SwittiPipeline:
 
     @classmethod
     def from_pretrained(cls, pretrained_model_name_or_path, device="cuda"):
-        switti = SwittiHF.from_pretrained(pretrained_model_name_or_path).to(device)
+        switti = SwittiHF.from_pretrained(pretrained_model_name_or_path, device=device).to(device)
         vae = VQVAEHF.from_pretrained(cls.vae_path).to(device)
         text_encoder = FrozenCLIPEmbedder(cls.text_encoder_path, device=device)
         text_encoder_2 = FrozenCLIPEmbedder(cls.text_encoder_2_path, device=device)