zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-12-16 02:08:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use model.safetensors with Whisper (#1399)
Some checks failed
Test / check_lint (push) Has been cancelled
Details

Browse Source
This commit is contained in:
Anthony
2025-12-15 15:17:08 +01:00
committed by GitHub
parent 7ddca42f4d
commit e52c128d11
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
whisper/mlx_whisper/load_models.py
View File

@@ -26,7 +26,10 @@ def load_model(
model_args = whisper.ModelDimensions(**config)
wf = model_path / "weights.safetensors"
# Prefer model.safetensors, fall back to weights.safetensors, then weights.npz
wf = model_path / "model.safetensors"
if not wf.exists():
wf = model_path / "weights.safetensors"
if not wf.exists():
wf = model_path / "weights.npz"
weights = mx.load(str(wf))