Skip to content
New issue

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

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

Already on GitHub? Sign in to your account

Add Gemini #2186

Merged
merged 7 commits into from
Dec 22, 2023
Merged

Add Gemini #2186

merged 7 commits into from
Dec 22, 2023

Conversation

JosselinSomervilleRoberts
Copy link
Contributor

This still needs more testing but so far works on MMLU.

@JosselinSomervilleRoberts
Copy link
Contributor Author

Should be good to go. The quota increase to 600 requests per minute has been submitted. I updated the metadata and parameters which I think are correct now.

# TODO: support VLM such as "gemini-pro-vision"
model_name: str = ""
if request.model_engine == "gemini":
model_name = "gemini-pro"
Copy link
Member

Choose a reason for hiding this comment

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

How does model versioning work? I believe sending a request with gemini-pro will just get the latest version.

Copy link
Member

Choose a reason for hiding this comment

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

We will revisit later when Google releases more versions.

src/helm/benchmark/test_model_properties.py Outdated Show resolved Hide resolved
src/helm/config/model_metadata.yaml Outdated Show resolved Hide resolved
@teetone
Copy link
Member

teetone commented Dec 22, 2023

I think it's simpler to just match their name and call it gemini-pro instead of gemini. It will make adding new versions of gemini-pro easier since we can just match the official names.

@teetone teetone merged commit 24b7f95 into main Dec 22, 2023
6 checks passed
@teetone teetone deleted the joss-gemini branch December 22, 2023 11:09
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.

2 participants