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 Gemma2 9B on Cloud Run example #113

Merged
merged 14 commits into from
Oct 29, 2024
Merged

Add Gemma2 9B on Cloud Run example #113

merged 14 commits into from
Oct 29, 2024

Conversation

alvarobartt
Copy link
Member

@alvarobartt alvarobartt commented Oct 16, 2024

Description

This PR adds an example on serving Gemma2 9B with TGI DLC on Cloud Run, to be used for a video recording demo on Cloud Run. The idea of this example is similar if not the same as in the Llama 3.1 8B example already created i.e. serving an AWQ quantized model to optimize the cold start while still maintaining the accuracy of the original model.

The Gemma2 9B AWQ quantized model is available in hugging-quants/gemma-2-9b-it-AWQ-INT4, under the hugging-quants organization which is managed by Hugging Face.

Additionally, this PR also updates the naming convention for the Cloud Run examples, since the tgi-deployment naming is vague, and since from now we may ship more examples, using a format as the following may make the most sense examples/cloud-run/deploy-<MODEL_ID>-on-cloud-run (or we could just remove the ending -on-cloud-run but that was for the sake of alignment with the examples/vertex-ai/notebooks/....

Rename required since most of the use-cases / examples are around the
`tgi-deployment` so the naming can be vague, and instead we're following
the same formatting for the naming as in `examples/vertex-ai/notebooks`
@HuggingFaceDocBuilderDev
Copy link
Collaborator

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@wietsevenema
Copy link
Contributor

@alvarobartt can you make sure this publishes this week? I added the link https://huggingface.co/docs/google-cloud/examples/cloud-run-deploy-gemma-2-on-cloud-run to the video description

@alvarobartt alvarobartt merged commit 183c9c8 into main Oct 29, 2024
2 checks passed
@alvarobartt alvarobartt deleted the gemma2-cloud-run branch October 29, 2024 07:43
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants