Spaces:

NeuralFalcon
/

VibeVoice-Colab

Running

NeuralFalcon commited on Aug 28

Commit

0ccb57a

verified ·

1 Parent(s): e9c2893

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -161,7 +161,7 @@ class VibeVoiceDemo:
         self.model = VibeVoiceForConditionalGenerationInference.from_pretrained(
             self.model_path,
             torch_dtype=torch.bfloat16,
-            device_map='cuda',
         )
         self.model.eval()
         self.model.model.noise_scheduler = self.model.model.noise_scheduler.from_config(

         self.model = VibeVoiceForConditionalGenerationInference.from_pretrained(
             self.model_path,
             torch_dtype=torch.bfloat16,
+            device_map=self.device,
         )
         self.model.eval()
         self.model.model.noise_scheduler = self.model.model.noise_scheduler.from_config(