Skip to content

feat: Amazon Bedrock support#7

Open
JGalego wants to merge 2 commits intogoogle-deepmind:mainfrom
JGalego:feat/bedrock
Open

feat: Amazon Bedrock support#7
JGalego wants to merge 2 commits intogoogle-deepmind:mainfrom
JGalego:feat/bedrock

Conversation

@JGalego
Copy link

@JGalego JGalego commented Aug 16, 2024

This PR adds support for Amazon Bedrock models (scorer + optimizer).

Example:

python optimize_instructions.py --optimizer="bedrock/mistral.mixtral-8x7b-instruct-v0:1" \
                                --scorer="bedrock/anthropic.claude-3-sonnet-20240229-v1:0" \
                                --instruction_pos="Q_begin" \
                                --dataset="gsm8k" \
                                --task="train"

@JGalego JGalego marked this pull request as ready for review August 16, 2024 02:29
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant