Daporte commited on
Commit
92745d0
·
1 Parent(s): b13e4db

Point the missing-checkpoint error at the model repo

Browse files

scripts/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.

Files changed (1) hide show
  1. 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). Fetch the weights with "
140
- "python scripts/fetch_models.py"
 
 
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: