From 892f4bd269370dfd93ae986639e850e5733e6337 Mon Sep 17 00:00:00 2001 From: Mattt Zmuda Date: Mon, 24 Jul 2023 03:33:20 -0700 Subject: [PATCH] Add information about training API to README --- README.md | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/README.md b/README.md index 497f486a..06b7436f 100644 --- a/README.md +++ b/README.md @@ -373,6 +373,14 @@ const response = await replicate.predictions.list(); ### `replicate.trainings.create` +Use the training API to fine-tune language models +to make them better at a particular task. +To see what **language models** currently support fine-tuning, +check out Replicate's [collection of trainable language models](https://replicate.com/collections/trainable-language-models). + +If you're looking to fine-tune **image models**, +check out Replicate's [guide to fine-tuning image models](https://replicate.com/docs/guides/fine-tune-an-image-model). + ```js const response = await replicate.trainings.create(model_owner, model_name, version_id, options); ``` @@ -404,6 +412,10 @@ const response = await replicate.trainings.create(model_owner, model_name, versi } ``` +> **Warning** +> If you try to fine-tune a model that doesn't support training, +> you'll get a `400 Bad Request` response from the server. + ### `replicate.trainings.get` ```js