Create Fine Tune — API

Query parameters

project_idQueryintegerrequired
Project ID
active_iamQueryintegeroptional
Active IAM ID (To access contact person account) Find your Active IAM ID here
locationQuerystringrequired
Location

Request body

application/json

object

namestringrequired

The name of the training job

training_typestringrequired

The type of training

training_inputsobjectrequired

Configuration for training inputs

isCloneboolean

model_namestringrequired

sku_item_price_idintegerrequired

sku_item_planstringrequired

sku_item_pricestringrequired

huggingface_integration_idintegerrequired

Responses

200Successful response with training job details.

object

codeinteger

HTTP status code for the operation.

example200

dataobject

Details of the training job.

errorsobject

Contains any errors that occurred.

messagestring

Result message confirming the operation.

exampleSuccess

Request

curl -X POST 'https://api.e2enetworks.com/myaccount/api/v1/gpu/finetuning/?project_id=%3Cproject_id%3E&active_iam=%3Cactive_iam%3E&location=%3Clocation%3E&apikey=%3Capikey%3E' \
  -H 'Authorization: Bearer <JWT>' \
  -H 'Content-Type: application/json' \
  -d '{
  "name": "string",
  "training_type": "string",
  "training_inputs": {
    "source_repository_type": "string",
    "dataset": "string",
    "dataset_type": "string",
    "validation_split_ratio": 0,
    "target_dataset_field": "string",
    "gradient_accumulation_steps": 0,
    "context_length": 0,
    "learning_rate": 0,
    "epochs": 0,
    "stop_training_when": "string",
    "max_steps": 0,
    "batch_size": 0,
    "peft_lora_alpha": 0,
    "peft_lora_r": 0,
    "max_grad_norm": 0,
    "save_strategy": "string",
    "save_steps": 0,
    "save_total_limit": 0,
    "lora_dropout": 0,
    "lora_bias": "string",
    "load_in_4bit": false,
    "image_column": "string",
    "caption_column": "string",
    "validation_prompt": "string",
    "num_validation_images": 0,
    "train_batch_size": 0,
    "load_in_8bit": false,
    "mixed_precision": "string",
    "checkpointing_steps": 0,
    "checkpoints_total_limit": 0,
    "task": "string",
    "prompt_configuration": "string"
  },
  "isClone": false,
  "model_name": "string",
  "sku_item_price_id": 0,
  "sku_item_plan": "string",
  "sku_item_price": "string",
  "huggingface_integration_id": 0
}'

Response · 200

{
  "code": 200,
  "data": {
    "id": 810,
    "name": "tir-finetuning-111511093030",
    "model_name": "stabilityai/stable-diffusion-2-1",
    "training_inputs": {
      "dataset": "lambdalabs/naruto-blip-captions",
      "dataset_name": "lambdalabs/naruto-blip-captions",
      "source_model_repo_info": "",
      "source_repository_type": "base_model",
      "source_model_path": "",
      "task": "Instruction-Finetuning",
      "dataset_type": "huggingface",
      "train_batch_size": 1,
      "gradient_accumulation_steps": 1,
      "epochs": 3,
      "max_steps": -1,
      "checkpointing_steps": 500,
      "checkpoints_total_limit": 10,
      "peft_lora_r": 64,
      "peft_lora_alpha": 16,
      "load_in_8bit": false,
      "mixed_precision": "no",
      "validation_prompt": "A photo of a man with green eyes",
      "num_validation_images": 2,
      "logging_steps": 5,
      "learning_rate": 0.0000141,
      "limit_training_records_count": -1,
      "caption_column": "text",
      "image_column": "image",
      "save_strategy": "steps",
      "stop_training_when": "epoch_count"
    },
    "description": null,
    "status": "Creating",
    "created_by": {
      "id": 3573,
      "name": "Nipun",
      "email": "nipun.arora@e2enetworks.com",
      "username": "nipun.arora@e2enetworks.com"
    },
    "created_at": "2024-11-15T06:10:06.278937Z",
    "hugging_face_integration_name": "tir",
    "sku_item_price": {
      "sku_item_price_id": 763,
      "sku_type": "hourly",
      "committed_days": 0,
      "unit_price": 470.4,
      "currency": "INR",
      "description": "Hourly Compute Instance: pipeline"
    },
    "sku": {
      "sku_id": 166,
      "name": "GDC.1xH200-30.375GB_SXM",
      "series": "GPU",
      "cpu": 30,
      "gpu": 1,
      "memory": 375,
      "is_free": false
    }
  },
  "errors": {},
  "message": "Success"
}