From e1545f4776ac11d1b006a63d1a1b5e930205c484 Mon Sep 17 00:00:00 2001 From: PiotrBLL Date: Mon, 11 Nov 2024 22:22:08 +0100 Subject: [PATCH] Fix: compute_device name in Whisper model --- whisper/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/whisper/__init__.py b/whisper/__init__.py index 3c027d7..55f461a 100644 --- a/whisper/__init__.py +++ b/whisper/__init__.py @@ -157,7 +157,7 @@ def load_model( del checkpoint_file dims = ModelDimensions(**checkpoint["dims"]) - model = Whisper(dims, device=torch.device(device)) + model = Whisper(dims, compute_device=torch.device(device)) model.load_state_dict(checkpoint["model_state_dict"]) if alignment_heads is not None: