Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Custom predictor: Taking a Ollama Custom Server for TinyLlama 1.1b for example #285

Merged
merged 8 commits into from
Feb 13, 2025

Conversation

madhurprash
Copy link
Collaborator

This PR contains the following additions:

  1. Adding a base config file and the custom predictor for the config file. This custom predictor is based on a custom payload.
  2. Adding a TinyLlama1.1b byoe file using the same custom response and payload format. Adding instructions to deploy the model here: https://github.com/madhurprash/EC2_tinyllama/tree/main, tested on the custom predictor.
  3. Comments throughout the config files.

@madhurprash madhurprash changed the title Custom predictor: Taking a Ollama Custom Server for TinyLlama 1.1b Custom predictor: Taking a Ollama Custom Server for TinyLlama 1.1b for example Feb 13, 2025
parameter_set: custom_rest
# This model id is appended to the prompt as the payload is sent to the model id
model_id: "<custom-model-id>"
headers:
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Add a comment in here that the keys in the header section are transparently passed via the POST request to the model server, so technically any key value pair you want to add in this section you can add.

@madhurprash
Copy link
Collaborator Author

updated

@aarora79 aarora79 merged commit e28af9b into aws-samples:main Feb 13, 2025
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants