É possível usar o SDK da Vertex AI para ajustar a maioria dos modelos que aceitam ajustes. Nem todos os modelos de fundação que aceitam isso podem ser ajustados usando o SDK da Vertex AI. Para ajustar um modelo, execute o método tune_model
no objeto do modelo. O método tune_model
permite ajustes supervisionados e não permite ajustes de aprendizado por reforço com feedback humano (RLHF).
Para mais detalhes sobre como usar a Vertex AI para ajustar um modelo, consulte Ajustar modelos de fundação de linguagem. Para ver um exemplo mais completo de como ajustar um modelo de geração de texto usando o SDK da Vertex AI, consulte Criar um job de ajuste supervisionado de modelo de texto.
Ajustar um modelo de fundação de geração de texto
As etapas a seguir demonstram como usar o SDK da Vertex AI para ajustar o modelo de fundação de geração de texto.
Chame o método
from_pretrained
para carregar uma versão do modelo de fundação de texto. O comando a seguir carrega a versão estável mais recente do modelo de fundação de geração de texto:my_text_generation_model = TextGenerationModel.from_pretrained("text-bison@002")
Execute o método
tune_model
para ajustar seu modelo de geração de texto.tune_model
inicia um job de ajuste de modelo e retorna imediatamente um objeto desse job. Para simplificar essa documentação, os parâmetros no snippet de código a seguir são substituídos por reticências (...
). Para saber mais sobre os parâmetros, consulte a definição do métodotune_model
na seção de modelo de linguagem do guia de referência do SDK da Vertex AI.tuning_job = my_text_generation_model.tune_model(...)
Execute o método
get_tuned_model
no job de ajuste para receber o modelo ajustado.get_tuned_model
aguarda a conclusão do job de ajuste iniciado portune_model
antes de retornar um modelo ajustado.tuned_model = tuning_job.get_tuned_model()
Depois de ajustar um modelo, use o método
list_tuned_model_names
para ver os modelos ajustados.list_tuned_model_names
retorna modelos ajustados do mesmo tipo de modelo que o objeto que o chamou. Por exemplo, se você chamarlist_tuned_model_names
no objeto de modeloTextGenerationModel.from_pretrained("text-bison@002")
, somente versões ajustadas do modelotext-bison@002
serão retornadas. O exemplo a seguir retorna os nomes dos modelos ajustados de geração de texto:text_generaton_model = TextGenerationModel.from_pretrained("text-bison@002") text_generaton_model.list_tuned_model_names()
A seguir
- Saiba como usar classes de modelo de texto e o SDK da Vertex AI.
- Saiba como usar classes de modelo de código e o SDK da Vertex AI.
- Saiba mais sobre classes do SDK da Vertex AI não relacionadas a IA generativa.