Update demos/musicgen_app.py

demos/musicgen_app.py

@@ -137,12 +137,7 @@ class Predictor:
         if self.is_daemon:
             # Running in a daemonic process (e.g., on Spaces)
             self.device = 'cuda' if torch.cuda.is_available() else 'cpu'
-            if depth=='float32':
-                self.model = MusicGen.get_pretrained(self.model_name, device=self.device)
-            if depth=='bfloat16':
-                self.model = MusicGen.get_pretrained(self.model_name, device=self.device, torch_dtype=torch.bfloat16)
-            if depth=='float16':
-                self.model = MusicGen.get_pretrained(self.model_name, device=self.device, torch_dtype=torch.float16)
+            self.model = MusicGen.get_pretrained(self.model_name, device=self.device, depth=depth)
             self.mbd = MultiBandDiffusion.get_mbd_musicgen(device=self.device)  # Load MBD here too
             self.current_task_id = 0 # Initialize task ID
         else: