I am using import google.generativeai as genai
library and creating tuning job but it is giving error:
google.api_core.exceptions.FailedPrecondition: 400 models/gemini-1.5-flash-001-tuning is not found for CREATE TUNED MODEL at API version v1beta.
Code:
genai.configure(api_key="")
with open("file.json", "r") as f:
training_dataset = json.load(f)
base_model_name = "models/gemini-1.5-flash-001-tuning"
operation = genai.create_tuned_model(
source_model=base_model_name,
training_data=training_dataset,
id="trial1-13-06-25",
epoch_count=5,
batch_size=8,
learning_rate=0.001,
)
print("Metadata: ", operation.metadata)