from seekrai.types import TrainingConfig, InfrastructureConfig
from seekrai import SeekrFlow
client = SeekrFlow()
training_config = TrainingConfig(
training_files=[
"file-830e9be3-25tt-13y1-0298-3a035e73o90" # Vision-language dataset file ID
],
model="meta-llama/Llama-3.2-11B-Vision-Instruct",
n_epochs=1,
n_checkpoints=1,
batch_size=4,
learning_rate=1e-5,
experiment_name="vlm_helperbot_v1",
)
infrastructure_config = InfrastructureConfig(
n_accel=8,
accel_type="MI300X",
)
fine_tune = client.fine_tuning.create(
training_config=training_config,
infrastructure_config=infrastructure_config,
project_id=123,
)
print(fine_tune.id)