from seekrai import SeekrFlow
from seekrai.types import TrainingConfig, InfrastructureConfig
from seekrai.types.finetune import LoRAConfig
client = SeekrFlow()
training_config = TrainingConfig(
training_files=["<your-training-file-id>"],
model="meta-llama/Meta-Llama-3-8B-Instruct",
n_epochs=1,
n_checkpoints=1,
batch_size=4,
learning_rate=1e-5,
experiment_name="helperbot_lora_v1",
lora_config=LoRAConfig(
r=16,
alpha=32,
),
)
infrastructure_config = InfrastructureConfig(
n_accel=8,
accel_type="GAUDI2",
)
fine_tune = client.fine_tuning.create(
training_config=training_config,
infrastructure_config=infrastructure_config,
project_id=123,
)
print(fine_tune.id)