basetenlabs / truss-examples

Examples of models deployable with Truss
https://trussml.com
MIT License
130 stars 37 forks source link

Fix mistral trt example. #278

Closed squidarth closed 5 months ago

squidarth commented 5 months ago

Summary

After mistral started requiring accepting terms, we need to start including the hf_access_token in secrets.

Note that for the TRT model, the code already sets the hf_access_token if it's in the secrets (it was generated from some template).

Note that I also added a "model_metadata.repo_id" field to the mistral-7b-chat and instruct models as well so that the terms and conditions link shows up properly.

Testing

Deployed the model on my account:

$ truss push --publish --wait --remote baseten --trusted