@civitai/client 0.2.0-beta.11 → 0.2.0-beta.14

package/dist/generated/sdk.gen.d.ts

@@ -23,6 +23,9 @@ import type {
   InvokeImageResourceTrainingStepTemplateData,
   InvokeImageResourceTrainingStepTemplateResponses,
   InvokeImageResourceTrainingStepTemplateErrors,
+  InvokeTrainingStepTemplateData,
+  InvokeTrainingStepTemplateResponses,
+  InvokeTrainingStepTemplateErrors,
   InvokeImageUploadStepTemplateData,
   InvokeImageUploadStepTemplateResponses,
   InvokeImageUploadStepTemplateErrors,
@@ -180,6 +183,18 @@ export declare const invokeImageResourceTrainingStepTemplate: <
   ThrowOnError,
   'fields'
 >;
+/**
+ * AI Toolkit Training
+ * Train models using AI Toolkit engine
+ */
+export declare const invokeTrainingStepTemplate: <ThrowOnError extends boolean = false>(
+  options?: Options<InvokeTrainingStepTemplateData, ThrowOnError>
+) => import('./client').RequestResult<
+  InvokeTrainingStepTemplateResponses,
+  InvokeTrainingStepTemplateErrors,
+  ThrowOnError,
+  'fields'
+>;
 /**
  * Image upload
  * Uploads an image to be used in a workflow

package/dist/generated/sdk.gen.js

@@ -154,6 +154,27 @@ export const invokeImageResourceTrainingStepTemplate = (options) => {
         },
     });
 };
+/**
+ * AI Toolkit Training
+ * Train models using AI Toolkit engine
+ */
+export const invokeTrainingStepTemplate = (options) => {
+    var _a;
+    return ((_a = options === null || options === void 0 ? void 0 : options.client) !== null && _a !== void 0 ? _a : _heyApiClient).post({
+        security: [
+            {
+                scheme: 'bearer',
+                type: 'http',
+            },
+        ],
+        url: '/v2/consumer/recipes/training',
+        ...options,
+        headers: {
+            'Content-Type': 'application/json',
+            ...options === null || options === void 0 ? void 0 : options.headers,
+        },
+    });
+};
 /**
  * Image upload
  * Uploads an image to be used in a workflow

package/dist/generated/types.gen.d.ts

@@ -628,6 +628,195 @@ export type ImageResourceTrainingStepTemplate = WorkflowStepTemplate & {
 } & {
   $type: 'imageResourceTraining';
 };
+/**
+ * Input for a training step.
+ */
+export type TrainingInput = {
+  /**
+   * The training engine to use
+   */
+  engine: 'ai-toolkit';
+  /**
+   * The model ecosystem
+   */
+  ecosystem: 'sd1' | 'sdxl' | 'sd3' | 'flux1' | 'wan';
+  /**
+   * The model URN to train upon
+   */
+  model: string;
+  /**
+   * Model variant. Required for: sd3 ('large'), flux1 ('dev' or 'schnell'), wan ('2.1' or '2.2'). Not used for sd1 or sdxl.
+   */
+  modelVariant?: string | null;
+  /**
+   * Training data specification
+   */
+  trainingData: {
+    /**
+     * Type of training data (e.g., 'zip')
+     */
+    type: string;
+    /**
+     * URN/URL to training data
+     */
+    sourceUrl: string;
+    /**
+     * Number of training items
+     */
+    count: number;
+  };
+  /**
+   * Sample generation configuration
+   */
+  samples: {
+    /**
+     * Sample generation prompts
+     */
+    prompts: Array<string>;
+  };
+  /**
+   * Number of training epochs
+   */
+  epochs?: number;
+  /**
+   * Training resolution (512, 1024, etc.)
+   */
+  resolution?: number | null;
+  /**
+   * Learning rate
+   */
+  lr?: number;
+  /**
+   * Text encoder learning rate
+   */
+  textEncoderLr?: number | null;
+  /**
+   * Whether to train text encoder
+   */
+  trainTextEncoder?: boolean;
+  /**
+   * Learning rate scheduler type
+   */
+  lrScheduler?: string;
+  /**
+   * Optimizer type
+   */
+  optimizerType?: string;
+  /**
+   * LoRA network dimension
+   */
+  networkDim?: number;
+  /**
+   * LoRA network alpha
+   */
+  networkAlpha?: number;
+  /**
+   * Noise offset for training
+   */
+  noiseOffset?: number;
+  /**
+   * Min SNR gamma value
+   */
+  minSnrGamma?: number | null;
+  /**
+   * Enable horizontal flip augmentation
+   */
+  flipAugmentation?: boolean;
+  /**
+   * Shuffle tokens during training
+   */
+  shuffleTokens?: boolean;
+  /**
+   * Number of tokens to keep at start
+   */
+  keepTokens?: number;
+};
+/**
+ * An epoch result for training output.
+ */
+export type TrainingEpochResult = {
+  /**
+   * The epoch number
+   */
+  epochNumber?: number;
+  /**
+   * The trained model blob for this epoch
+   */
+  model: {
+    /**
+     * The blob id of the trained model file
+     */
+    id: string;
+    /**
+     * Whether the model blob is available
+     */
+    available: boolean;
+    /**
+     * Presigned URL to download the model
+     */
+    url?: string | null;
+    /**
+     * When the presigned URL expires
+     */
+    urlExpiresAt?: string | null;
+  };
+  /**
+   * Generated sample images for this epoch
+   */
+  samples: Array<{
+    /**
+     * The blob id of the sample image
+     */
+    id: string;
+    /**
+     * Whether the sample image is available
+     */
+    available: boolean;
+    /**
+     * Presigned URL to view the sample image
+     */
+    url?: string | null;
+    /**
+     * When the presigned URL expires
+     */
+    urlExpiresAt?: string | null;
+    /**
+     * NSFW level of the sample image
+     */
+    nsfwLevel?: string | null;
+  }>;
+};
+export type TrainingOutput = {
+  /**
+   * Moderation status of the training output
+   */
+  moderationStatus: string;
+  /**
+   * Training epoch results
+   */
+  epochs?: Array<TrainingEpochResult> | null;
+};
+/**
+ * AI Toolkit Training Step
+ */
+export type TrainingStep = WorkflowStep & {
+  $type: 'training';
+} & {
+  input: TrainingInput;
+  output?: TrainingOutput;
+} & {
+  $type: 'training';
+};
+/**
+ * AI Toolkit Training Step Template
+ */
+export type TrainingStepTemplate = WorkflowStepTemplate & {
+  $type: 'training';
+} & {
+  input: TrainingInput;
+} & {
+  $type: 'training';
+};
 /**
  * Available image transformers.
  */
@@ -2489,6 +2678,35 @@ export type InvokeImageResourceTrainingStepTemplateResponses = {
 };
 export type InvokeImageResourceTrainingStepTemplateResponse =
   InvokeImageResourceTrainingStepTemplateResponses[keyof InvokeImageResourceTrainingStepTemplateResponses];
+export type InvokeTrainingStepTemplateData = {
+  body?: TrainingInput;
+  path?: never;
+  query?: {
+    experimental?: boolean;
+    allowMatureContent?: boolean;
+  };
+  url: '/v2/consumer/recipes/training';
+};
+export type InvokeTrainingStepTemplateErrors = {
+  /**
+   * Bad Request
+   */
+  400: ProblemDetails;
+  /**
+   * Unauthorized
+   */
+  401: ProblemDetails;
+};
+export type InvokeTrainingStepTemplateError =
+  InvokeTrainingStepTemplateErrors[keyof InvokeTrainingStepTemplateErrors];
+export type InvokeTrainingStepTemplateResponses = {
+  /**
+   * OK
+   */
+  200: TrainingOutput;
+};
+export type InvokeTrainingStepTemplateResponse =
+  InvokeTrainingStepTemplateResponses[keyof InvokeTrainingStepTemplateResponses];
 export type InvokeImageUploadStepTemplateData = {
   body?: string;
   path?: never;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@civitai/client",
-  "version": "0.2.0-beta.11",
+  "version": "0.2.0-beta.14",
   "description": "Civitai's javascript client for generating ai content",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -54,4 +54,4 @@
   "publishConfig": {
     "access": "public"
   }
-}
+}