inference: Fix download command in error msg (#133)

I got this error message and tried to the run the command presented
and it didn't work. The model needs to be give with `--model-id`
instead of as a positional argument.

Signed-off-by: Russell Bryant <rbryant@redhat.com>
This commit is contained in:
Russell Bryant 2024-09-27 16:31:11 -04:00 committed by GitHub
parent fb9e6371ec
commit 5828ffd53b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -53,7 +53,7 @@ def model_checkpoint_dir(model) -> str:
assert checkpoint_dir.exists(), ( assert checkpoint_dir.exists(), (
f"Could not find checkpoint dir: {checkpoint_dir}." f"Could not find checkpoint dir: {checkpoint_dir}."
f"Please download model using `llama download {model.descriptor()}`" f"Please download model using `llama download --model-id {model.descriptor()}`"
) )
return str(checkpoint_dir) return str(checkpoint_dir)