By using AWS re:Post, you agree to the AWS re:Post Terms of Use
AWS re:Postre:Post

Could we use the Hugging face model weights in Sagemaker?

0

On Google colab, I'm fine tuning a stable diffusion model and deploying it on hugging faces. Can I use these weights in Sagamaker? Can you give me a reference link on how to use this?

Topics
Machine Learning & AI
Tags
Build & Train ML ModelsAmazon SageMakerMachine Learning & AIAmazon SageMaker Deployment
Language
English
Dayanand
asked 8 months ago291 views
1 Answer
0

You can deploy model with weights using the LMI container that sagemaker provides. Please refer the below example https://github.com/aws-samples/amazon-sagemaker-generativeai/blob/main/instruction-fine-tuning/gpt-neo/djl_deploy_GPT_Neo.ipynb

AWS
AWS-User-8567723
answered 8 months ago

You are not logged in. Log in to post an answer.

A good answer clearly answers the question and provides constructive feedback and encourages professional growth in the question asker.

Guidelines for Answering Questions

Relevant content