Point the missing-checkpoint error at the model repo
Browse filesscripts/fetch_models.py does not exist in either published repo, so anyone
who hit this error was sent to a dead command. The weights arrive through
from_pretrained now.
- whisper_decoder.py +4 -2
whisper_decoder.py
CHANGED
|
@@ -136,8 +136,10 @@ class Whisper:
|
|
| 136 |
if not (model_dir / "config.json").exists():
|
| 137 |
raise FileNotFoundError(
|
| 138 |
f"{model_dir} is not a Whisper checkpoint directory "
|
| 139 |
-
"(no config.json).
|
| 140 |
-
"
|
|
|
|
|
|
|
| 141 |
)
|
| 142 |
|
| 143 |
if device is None:
|
|
|
|
| 136 |
if not (model_dir / "config.json").exists():
|
| 137 |
raise FileNotFoundError(
|
| 138 |
f"{model_dir} is not a Whisper checkpoint directory "
|
| 139 |
+
"(no config.json). The weights come from the model repo, so "
|
| 140 |
+
"load through it rather than pointing here by hand:\n"
|
| 141 |
+
" AutoModel.from_pretrained("
|
| 142 |
+
"'Daporte/prosody2text-weights', trust_remote_code=True)"
|
| 143 |
)
|
| 144 |
|
| 145 |
if device is None:
|