---
title: "Update Training Shape"
method: PATCH
path: "/v1/accounts/{account_id}/trainingShapes/{training_shape_id}"
tags: ["gateway.openapi_Gateway"]
---

# Update Training Shape

`PATCH /v1/accounts/{account_id}/trainingShapes/{training_shape_id}`

## Request

- Base URL: `https://api.fireworks.ai`
- URL: `https://api.fireworks.ai/v1/accounts/{account_id}/trainingShapes/{training_shape_id}`
- Auth: HTTP bearer
- TypeScript: <https://skmtc.dev/fw-ai-external/apis/gateway-rest-api/docs/v1/accounts/:account_id/trainingShapes/:training_shape_id/patch.ts> (a function with typed inputs; its response is checked against the schema)

## Path parameters

- `account_id` string, required
- `training_shape_id` string, required

## Request body

- object
  - `displayName` string: Human-readable display name of the training shape. e.g. "Llama3 70B H200 BF16" Must be fewer than 64 characters long.
  - `description` string: The description of the training shape. Must be fewer than 1000 characters long.
  - `createTime` string, date-time: The creation time of the training shape.
  - `updateTime` string, date-time: The update time for the training shape.
  - `baseModel` string, required
  - `deploymentShapeVersion` string
  - `trainerImageTag` string, required: The validated trainer runtime image tag used for numerics verification.
  - `trainerMode` 'TRAINER_MODE_UNSPECIFIED' | 'POLICY_TRAINER' | 'FORWARD_ONLY' | 'LORA_TRAINER': Trainer execution mode used for validated launch-profile matching.
  - `nodeCount` integer: Node count validated for the launch profile.
  - `trainerShardingScheme` GatewayTrainerShardingScheme: Structured parallelism/sharding profile used by trainer launches.
    - `tensorParallelism` integer: Tensor-parallel degree. 0 means unspecified (server defaults to 1).
    - `pipelineParallelism` integer: Pipeline-parallel degree. 0 means unspecified (server defaults to 1).
    - `contextParallelism` integer: Context-parallel degree. 0 means unspecified (server defaults to 1).
    - `expertParallelism` integer: Expert-parallel degree. 0 means unspecified (server defaults to 1).
    - `sequenceParallelism` boolean: Whether sequence parallelism should be enabled.
  - `modelType` string: The model type of the base model (e.g. "llama", "qwen").
  - `parameterCount` string, int64: The parameter count of the base model, used for billion-parameter bucket matching.
  - `acceleratorType` 'ACCELERATOR_TYPE_UNSPECIFIED' | 'NVIDIA_A100_80GB' | 'NVIDIA_H100_80GB' | 'AMD_MI300X_192GB' | 'NVIDIA_A10G_24GB' | 'NVIDIA_A100_40GB' | 'NVIDIA_L4_24GB' | 'NVIDIA_H200_141GB' | 'NVIDIA_B200_180GB' | 'AMD_MI325X_256GB' | 'AMD_MI350X_288GB' | 'NVIDIA_B300_288GB' | 'NVIDIA_GB200' | 'NVIDIA_GB300'
  - `acceleratorCount` integer: Total number of accelerators used by the job.
  - `baseModelWeightPrecision` 'WEIGHT_PRECISION_UNSPECIFIED' | 'BFLOAT16' | 'INT8' | 'NF4' | 'FP8' | 'FP4_FP8': The weight precision for model training/inference. - BFLOAT16: no quantization applied - INT8: enable 8-bit quantization with LLM.int8() - NF4: enable 4-bit quantization with LLM.nf4() - FP8: base model quantization in FP8 - FP4_FP8: base model linear module quantization in FP4, mixed with experts and some special keys/layers in FP8.
  - `maxSupportedContextLength` integer: Capacity limits validated for this shape.
  - `minTotalAcceleratorCount` integer: Minimum total accelerator count (world size, across all data-parallel replicas) this shape has been validated at. Written from shape-validation CI evidence; do not hand-edit downward.

## Response `200`

A successful response.

- GatewayTrainingShape
  - `name` string
  - `displayName` string: Human-readable display name of the training shape. e.g. "Llama3 70B H200 BF16" Must be fewer than 64 characters long.
  - `description` string: The description of the training shape. Must be fewer than 1000 characters long.
  - `createTime` string, date-time: The creation time of the training shape.
  - `updateTime` string, date-time: The update time for the training shape.
  - `baseModel` string, required
  - `deploymentShapeVersion` string
  - `trainerImageTag` string, required: The validated trainer runtime image tag used for numerics verification.
  - `trainerMode` 'TRAINER_MODE_UNSPECIFIED' | 'POLICY_TRAINER' | 'FORWARD_ONLY' | 'LORA_TRAINER': Trainer execution mode used for validated launch-profile matching.
  - `nodeCount` integer: Node count validated for the launch profile.
  - `trainerShardingScheme` GatewayTrainerShardingScheme: Structured parallelism/sharding profile used by trainer launches.
    - `tensorParallelism` integer: Tensor-parallel degree. 0 means unspecified (server defaults to 1).
    - `pipelineParallelism` integer: Pipeline-parallel degree. 0 means unspecified (server defaults to 1).
    - `contextParallelism` integer: Context-parallel degree. 0 means unspecified (server defaults to 1).
    - `expertParallelism` integer: Expert-parallel degree. 0 means unspecified (server defaults to 1).
    - `sequenceParallelism` boolean: Whether sequence parallelism should be enabled.
  - `modelType` string: The model type of the base model (e.g. "llama", "qwen").
  - `parameterCount` string, int64: The parameter count of the base model, used for billion-parameter bucket matching.
  - `acceleratorType` 'ACCELERATOR_TYPE_UNSPECIFIED' | 'NVIDIA_A100_80GB' | 'NVIDIA_H100_80GB' | 'AMD_MI300X_192GB' | 'NVIDIA_A10G_24GB' | 'NVIDIA_A100_40GB' | 'NVIDIA_L4_24GB' | 'NVIDIA_H200_141GB' | 'NVIDIA_B200_180GB' | 'AMD_MI325X_256GB' | 'AMD_MI350X_288GB' | 'NVIDIA_B300_288GB' | 'NVIDIA_GB200' | 'NVIDIA_GB300'
  - `acceleratorCount` integer: Total number of accelerators used by the job.
  - `baseModelWeightPrecision` 'WEIGHT_PRECISION_UNSPECIFIED' | 'BFLOAT16' | 'INT8' | 'NF4' | 'FP8' | 'FP4_FP8': The weight precision for model training/inference. - BFLOAT16: no quantization applied - INT8: enable 8-bit quantization with LLM.int8() - NF4: enable 4-bit quantization with LLM.nf4() - FP8: base model quantization in FP8 - FP4_FP8: base model linear module quantization in FP4, mixed with experts and some special keys/layers in FP8.
  - `maxSupportedContextLength` integer: Capacity limits validated for this shape.
  - `minTotalAcceleratorCount` integer: Minimum total accelerator count (world size, across all data-parallel replicas) this shape has been validated at. Written from shape-validation CI evidence; do not hand-edit downward.

## Changes

- **2026-10-16** `3bf245180173`: 1 info
  - endpoint added

[Operation history](https://skmtc.dev/fw-ai-external/apis/gateway-rest-api/changes/v1/accounts/:account_id/trainingShapes/:training_shape_id/patch.md)

---

[API](https://skmtc.dev/fw-ai-external/apis/gateway-rest-api.md) - [All operations](https://skmtc.dev/fw-ai-external/apis/gateway-rest-api/llms.txt) - [OpenAPI document](https://skmtc.dev/fw-ai-external/apis/gateway-rest-api/revisions/3bf245180173?raw)
