Update modeling.py

modeling.py

@@ -98,7 +98,7 @@ class VoiceRestore(PreTrainedModel):
 
             # Restore audio
             with torch.no_grad():
-                with torch.autocast(device):
+                with torch.autocast(device.type):
                     restored_mel = model.voice_restore.sample(processed_mel.transpose(1, 2), steps=steps, cfg_strength=cfg_strength)
                     restored_mel = restored_mel.squeeze(0).transpose(0, 1)