Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-11-24 06:26:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix: compute_device name in Whisper model

Browse Source
This commit is contained in:
PiotrBLL 2024-11-11 22:22:08 +01:00
parent b0cf21b9b5
commit e1545f4776
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
whisper/__init__.py
View File

@ -157,7 +157,7 @@ def load_model(
del checkpoint_file del checkpoint_file
dims = ModelDimensions(**checkpoint["dims"]) 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"]) model.load_state_dict(checkpoint["model_state_dict"])
if alignment_heads is not None: if alignment_heads is not None: