Menggunakan Vertex AI SDK untuk menyesuaikan model

Anda dapat menggunakan Vertex AI SDK untuk menyesuaikan sebagian besar model yang mendukung penyesuaian. Tidak semua model dasar yang mendukung penyesuaian dapat disesuaikan menggunakan Vertex AI SDK. Untuk menyesuaikan model, jalankan metode tune_model pada objek model. Metode tune_model mendukung penyesuaian yang diawasi dan tidak mendukung penyesuaian reinforcement learning from human feedback (RLHF). Untuk mengetahui detail tentang penggunaan Vertex AI untuk menyesuaikan model, lihat Menyesuaikan model dasar bahasa. Untuk contoh yang lebih lengkap tentang cara menyesuaikan model pembuatan teks menggunakan Vertex AI SDK, lihat Membuat tugas penyesuaian model teks yang diawasi.

Menyesuaikan model dasar pembuatan teks

Langkah berikut menunjukkan cara menggunakan Vertex AI SDK untuk menyesuaikan model dasar pembuatan teks.

  1. Panggil metode from_pretrained untuk memuat versi model dasar teks. Kode berikut memuat versi stabil terbaru dari model dasar pembuatan teks:

    my_text_generation_model = TextGenerationModel.from_pretrained("text-bison@002")

  2. Jalankan metode tune_model untuk menyesuaikan model pembuatan teks Anda. tune_model meluncurkan tugas penyesuaian model, lalu segera menampilkan objek tugas tuning. Untuk menyederhanakan dokumentasi ini, parameter dalam cuplikan kode berikut diganti dengan elipsis (...). Untuk mempelajari parameter, lihat definisi metode tune_model di bagian model bahasa dalam panduan referensi Vertex AI SDK.

    tuning_job = my_text_generation_model.tune_model(...)

  3. Jalankan metode get_tuned_model pada tugas tuning untuk mendapatkan model yang telah disesuaikan. get_tuned_model menunggu tugas penyesuaian yang dimulai oleh tune_model selesai sebelum menampilkan model yang telah disesuaikan.

    tuned_model = tuning_job.get_tuned_model()

  4. Setelah Anda menyesuaikan model, gunakan metode list_tuned_model_names untuk melihat model yang telah disesuaikan. list_tuned_model_names menampilkan model yang disesuaikan dan memiliki jenis model yang sama dengan objek yang memanggilnya. Misalnya, jika Anda memanggil list_tuned_model_names pada objek model TextGenerationModel.from_pretrained("text-bison@002"), hanya versi yang telah disesuaikan dari model text-bison@002 yang akan ditampilkan. Kode berikut menampilkan nama model pembuatan teks yang disesuaikan:

    text_generaton_model = TextGenerationModel.from_pretrained("text-bison@002")
text_generaton_model.list_tuned_model_names()

Langkah selanjutnya