@civitai/client 0.1.9-beta.24 → 0.1.9-beta.26

package/dist/client/CivitaiClient.d.ts

@@ -1,5 +1,6 @@
 type ClientConfig = {
   env?: 'dev' | 'prod';
+  baseUrl?: string;
   base?: string;
   auth: string;
 };

package/dist/client/CivitaiClient.js

@@ -1,9 +1,10 @@
 import { createClient } from '@hey-api/client-fetch';
 export function createCivitaiClient(config) {
     const client = createClient({
-        baseUrl: config.env === 'dev'
-            ? 'https://orchestration-dev.civitai.com'
-            : 'https://orchestration.civitai.com',
+        baseUrl: config.baseUrl ? config.baseUrl :
+            config.env === 'dev'
+                ? 'https://orchestration-dev.civitai.com'
+                : 'https://orchestration.civitai.com',
         global: false,
         headers: {
             Authorization: `Bearer ${config.auth}`,

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

@@ -1216,6 +1216,7 @@ export declare const $ImageResourceTrainingInput: {
     readonly mapping: {
       readonly kohya: '#/components/schemas/KohyaImageResourceTrainingInput';
       readonly 'flux-dev-fast': '#/components/schemas/FluxDevFastImageResourceTrainingInput';
+      readonly musubi: '#/components/schemas/MusubiImageResourceTrainingInput';
     };
   };
 };
@@ -1525,6 +1526,7 @@ export declare const $Job: {
       readonly textToImage: '#/components/schemas/TextToImageJob';
       readonly textToImageV2: '#/components/schemas/TextToImageV2Job';
       readonly imageResourceTraining: '#/components/schemas/ImageResourceTrainingJob';
+      readonly wdTagging: '#/components/schemas/WDTaggingJob';
       readonly comfy: '#/components/schemas/ComfyJob';
       readonly similaritySearch: '#/components/schemas/SimilaritySearchJob';
       readonly llmPromptAugmentation: '#/components/schemas/LLMPromptAugmentationJob';
@@ -1805,7 +1807,7 @@ export declare const $KohyaImageResourceTrainingInput: {
           readonly maximum: 3;
           readonly minimum: 0;
           readonly type: 'integer';
-          readonly description: 'If your training images have tags, you can randomly shuffle them. \nHowever, if you have words that you want to keep at the beginning, you can use this option to specify "Keep the first 0 words at the beginning".\nThis option does nothing if the Shuffle Tags option is off.';
+          readonly description: 'If your training images have tags, you can randomly shuffle them.\nHowever, if you have words that you want to keep at the beginning, you can use this option to specify "Keep the first 0 words at the beginning".\nThis option does nothing if the Shuffle Tags option is off.';
           readonly format: 'int32';
           readonly default: 0;
         };
@@ -1855,7 +1857,7 @@ export declare const $KohyaImageResourceTrainingInput: {
           readonly maximum: 20;
           readonly minimum: 0;
           readonly type: 'integer';
-          readonly description: 'Learning is performed by putting noise of various strengths on the training image, \nbut depending on the difference in strength of the noise on which it is placed, learning will be\nstable by moving closer to or farther from the learning target.\n\nMin SNR gamma was introduced to compensate for that. When learning images have little noise,\nit may deviate greatly from the target, so try to suppress this jump.';
+          readonly description: 'Learning is performed by putting noise of various strengths on the training image,\nbut depending on the difference in strength of the noise on which it is placed, learning will be\nstable by moving closer to or farther from the learning target.\n            \nMin SNR gamma was introduced to compensate for that. When learning images have little noise,\nit may deviate greatly from the target, so try to suppress this jump.';
           readonly format: 'int32';
           readonly nullable: true;
         };
@@ -1871,7 +1873,7 @@ export declare const $KohyaImageResourceTrainingInput: {
           readonly maximum: 256;
           readonly minimum: 1;
           readonly type: 'integer';
-          readonly description: 'The smaller the Network alpha value, the larger the stored LoRA neural net weights. \nFor example, with an Alpha of 16 and a Dim of 32, the strength of the weight used is 16/32 = 0.5, \nmeaning that the learning rate is only half as powerful as the Learning Rate setting.\n\nIf Alpha and Dim are the same number, the strength used will be 1 and will have no effect on the learning rate.';
+          readonly description: 'The smaller the Network alpha value, the larger the stored LoRA neural net weights.\nFor example, with an Alpha of 16 and a Dim of 32, the strength of the weight used is 16/32 = 0.5,\nmeaning that the learning rate is only half as powerful as the Learning Rate setting.\n            \nIf Alpha and Dim are the same number, the strength used will be 1 and will have no effect on the learning rate.';
           readonly format: 'int32';
           readonly nullable: true;
         };
@@ -1885,7 +1887,7 @@ export declare const $KohyaImageResourceTrainingInput: {
         };
         readonly optimizerType: {
           readonly type: 'string';
-          readonly description: 'The optimizer determines how to update the neural net weights during training. \nVarious methods have been proposed for smart learning, but the most commonly used in LoRA learning \nis "AdamW8bit" or "Adafactor" for SDXL.';
+          readonly description: 'The optimizer determines how to update the neural net weights during training.\nVarious methods have been proposed for smart learning, but the most commonly used in LoRA learning\nis "AdamW8bit" or "Adafactor" for SDXL.';
           readonly nullable: true;
         };
         readonly targetSteps: {
@@ -2137,6 +2139,112 @@ export declare const $MochiVideoGenJob: {
     };
   };
 };
+export declare const $MusubiImageResourceTrainingInput: {
+  readonly required: readonly ['engine'];
+  readonly allOf: readonly [
+    {
+      readonly $ref: '#/components/schemas/ImageResourceTrainingInput';
+    },
+    {
+      readonly type: 'object';
+      readonly properties: {
+        readonly maxTrainEpochs: {
+          readonly maximum: 20;
+          readonly minimum: 0;
+          readonly type: 'integer';
+          readonly description: 'An epoch is one set of learning. By default, we will save a maximum of 20 epochs (evenly distributed), and they are all available for download.';
+          readonly format: 'int32';
+          readonly default: 5;
+        };
+        readonly numRepeats: {
+          readonly maximum: 5000;
+          readonly minimum: 1;
+          readonly type: 'integer';
+          readonly description: 'Num Repeats defines how many times each individual image gets put into VRAM. As opposed to batch size, which is how many images are placed into VRAM at once.';
+          readonly format: 'int32';
+          readonly default: 8;
+        };
+        readonly trainBatchSize: {
+          readonly maximum: 9;
+          readonly minimum: 1;
+          readonly type: 'integer';
+          readonly description: 'Batch size is the number of images that will be placed into VRAM at once. A batch size of 2 will train two images at a time, simultaneously.';
+          readonly format: 'int32';
+          readonly nullable: true;
+        };
+        readonly resolution: {
+          readonly maximum: 1024;
+          readonly minimum: 512;
+          readonly type: 'integer';
+          readonly description: 'Specify the maximum resolution of training images. If the training images exceed the resolution specified here, they will be scaled down to this resolution';
+          readonly format: 'int32';
+          readonly nullable: true;
+        };
+        readonly enableBucket: {
+          readonly type: 'boolean';
+          readonly description: 'Sorts images into buckets by size for the purposes of training. If your training images are all the same size, you can turn this option off, but leaving it on has no effect.';
+          readonly default: true;
+        };
+        readonly unetLR: {
+          readonly maximum: 1;
+          readonly minimum: 0;
+          readonly type: 'number';
+          readonly description: 'Sets the learning rate for U-Net. This is the learning rate when performing additional learning on each attention block (and other blocks depending on the setting) in U-Net';
+          readonly format: 'double';
+          readonly default: 0.00005;
+        };
+        readonly lrScheduler: {
+          readonly enum: readonly ['constant', 'cosine', 'cosine_with_restarts', 'linear'];
+          readonly type: 'string';
+          readonly description: 'You can change the learning rate in the middle of learning. A scheduler is a setting for how to change the learning rate.';
+          readonly nullable: true;
+        };
+        readonly lrSchedulerNumCycles: {
+          readonly maximum: 4;
+          readonly minimum: 1;
+          readonly type: 'integer';
+          readonly description: 'This option specifies how many cycles the scheduler runs during training. It is only used when "cosine_with_restarts" or "polynomial" is used as the scheduler.';
+          readonly format: 'int32';
+          readonly default: 3;
+        };
+        readonly networkDim: {
+          readonly maximum: 256;
+          readonly minimum: 1;
+          readonly type: 'integer';
+          readonly description: 'The larger the Dim setting, the more learning information can be stored, but the possibility of learning unnecessary information other than the learning target increases. A larger Dim also increases LoRA file size.';
+          readonly format: 'int32';
+          readonly nullable: true;
+        };
+        readonly networkAlpha: {
+          readonly maximum: 256;
+          readonly minimum: 1;
+          readonly type: 'integer';
+          readonly description: 'The smaller the Network alpha value, the larger the stored LoRA neural net weights.\nFor example, with an Alpha of 16 and a Dim of 32, the strength of the weight used is 16/32 = 0.5,\nmeaning that the learning rate is only half as powerful as the Learning Rate setting.\n            \nIf Alpha and Dim are the same number, the strength used will be 1 and will have no effect on the learning rate.';
+          readonly format: 'int32';
+          readonly nullable: true;
+        };
+        readonly optimizerType: {
+          readonly type: 'string';
+          readonly description: 'The optimizer determines how to update the neural net weights during training.\nVarious methods have been proposed for smart learning, but the most commonly used in LoRA learning\nis "AdamW8bit" or "Adafactor" for SDXL.';
+          readonly nullable: true;
+        };
+        readonly targetSteps: {
+          readonly type: 'integer';
+          readonly format: 'int32';
+          readonly nullable: true;
+          readonly readOnly: true;
+        };
+      };
+      readonly additionalProperties: false;
+    },
+  ];
+  readonly properties: {
+    readonly engine: {
+      readonly enum: readonly ['musubi'];
+      readonly type: 'string';
+    };
+  };
+};
 export declare const $NSFWLevel: {
   readonly enum: readonly ['pg', 'pG13', 'r', 'x', 'xxx', 'na'];
   readonly type: 'string';
@@ -3341,6 +3449,55 @@ export declare const $ViduVideoGenStyle: {
   readonly enum: readonly ['general', 'anime'];
   readonly type: 'string';
 };
+export declare const $WDTaggingJob: {
+  readonly required: readonly ['$type'];
+  readonly allOf: readonly [
+    {
+      readonly $ref: '#/components/schemas/Job';
+    },
+    {
+      readonly type: 'object';
+      readonly properties: {
+        readonly model: {
+          readonly type: 'string';
+        };
+        readonly mediaUrl: {
+          readonly type: 'string';
+          readonly format: 'uri';
+        };
+        readonly threshold: {
+          readonly type: 'number';
+          readonly format: 'double';
+          readonly nullable: true;
+        };
+        readonly movieRatingModel: {
+          readonly type: 'string';
+          readonly nullable: true;
+        };
+        readonly prompt: {
+          readonly type: 'string';
+          readonly nullable: true;
+        };
+        readonly claimDuration: {
+          readonly type: 'string';
+          readonly format: 'date-span';
+          readonly readOnly: true;
+        };
+        readonly type: {
+          readonly type: 'string';
+          readonly readOnly: true;
+        };
+      };
+      readonly additionalProperties: false;
+    },
+  ];
+  readonly properties: {
+    readonly $type: {
+      readonly enum: readonly ['wdTagging'];
+      readonly type: 'string';
+    };
+  };
+};
 export declare const $WorkerCapabilities: {
   readonly type: 'object';
   readonly properties: {
@@ -3796,6 +3953,12 @@ export declare const $WorkerRegistration: {
       readonly description: 'Get additional metadata about this worker. This can be used for debugging purposes as well as to target certain configurations to certain workers.';
       readonly nullable: true;
     };
+    readonly capacitySaturationRate: {
+      readonly type: 'number';
+      readonly description: 'The saturation rate of available capacity of this worker of the worker.';
+      readonly format: 'double';
+      readonly nullable: true;
+    };
   };
   readonly additionalProperties: false;
   readonly description: "Details of a worker's registration.";
@@ -4115,6 +4278,12 @@ export declare const $WorkflowStep: {
       readonly additionalProperties: {};
       readonly description: 'A collection of user defined metadata for the workflow step.';
     };
+    readonly estimatedProgressRate: {
+      readonly type: 'number';
+      readonly description: 'An estimation on the current progression of this step, or null if there is no estimation';
+      readonly format: 'double';
+      readonly nullable: true;
+    };
   };
   readonly additionalProperties: false;
   readonly description: 'Details of a workflow step.';
@@ -4185,6 +4354,12 @@ export declare const $WorkflowStepJob: {
       readonly description: "The job's cost.";
       readonly format: 'double';
     };
+    readonly estimatedProgressRate: {
+      readonly type: 'number';
+      readonly description: 'An estimation on the current progression of this job, or null if there is no estimation';
+      readonly format: 'double';
+      readonly nullable: true;
+    };
   };
   readonly additionalProperties: false;
   readonly description: 'Details of a job produced by a workflow step.';

package/dist/generated/schemas.gen.js

@@ -1204,6 +1204,7 @@ export const $ImageResourceTrainingInput = {
         mapping: {
             kohya: '#/components/schemas/KohyaImageResourceTrainingInput',
             'flux-dev-fast': '#/components/schemas/FluxDevFastImageResourceTrainingInput',
+            musubi: '#/components/schemas/MusubiImageResourceTrainingInput',
         },
     },
 };
@@ -1515,6 +1516,7 @@ export const $Job = {
             textToImage: '#/components/schemas/TextToImageJob',
             textToImageV2: '#/components/schemas/TextToImageV2Job',
             imageResourceTraining: '#/components/schemas/ImageResourceTrainingJob',
+            wdTagging: '#/components/schemas/WDTaggingJob',
             comfy: '#/components/schemas/ComfyJob',
             similaritySearch: '#/components/schemas/SimilaritySearchJob',
             llmPromptAugmentation: '#/components/schemas/LLMPromptAugmentationJob',
@@ -1795,7 +1797,7 @@ export const $KohyaImageResourceTrainingInput = {
                     maximum: 3,
                     minimum: 0,
                     type: 'integer',
-                    description: `If your training images have tags, you can randomly shuffle them. 
+                    description: `If your training images have tags, you can randomly shuffle them.
 However, if you have words that you want to keep at the beginning, you can use this option to specify "Keep the first 0 words at the beginning".
 This option does nothing if the Shuffle Tags option is off.`,
                     format: 'int32',
@@ -1847,10 +1849,10 @@ This option does nothing if the Shuffle Tags option is off.`,
                     maximum: 20,
                     minimum: 0,
                     type: 'integer',
-                    description: `Learning is performed by putting noise of various strengths on the training image, 
+                    description: `Learning is performed by putting noise of various strengths on the training image,
 but depending on the difference in strength of the noise on which it is placed, learning will be
 stable by moving closer to or farther from the learning target.
-
+            
 Min SNR gamma was introduced to compensate for that. When learning images have little noise,
 it may deviate greatly from the target, so try to suppress this jump.`,
                     format: 'int32',
@@ -1868,10 +1870,10 @@ it may deviate greatly from the target, so try to suppress this jump.`,
                     maximum: 256,
                     minimum: 1,
                     type: 'integer',
-                    description: `The smaller the Network alpha value, the larger the stored LoRA neural net weights. 
-For example, with an Alpha of 16 and a Dim of 32, the strength of the weight used is 16/32 = 0.5, 
+                    description: `The smaller the Network alpha value, the larger the stored LoRA neural net weights.
+For example, with an Alpha of 16 and a Dim of 32, the strength of the weight used is 16/32 = 0.5,
 meaning that the learning rate is only half as powerful as the Learning Rate setting.
-
+            
 If Alpha and Dim are the same number, the strength used will be 1 and will have no effect on the learning rate.`,
                     format: 'int32',
                     nullable: true,
@@ -1886,8 +1888,8 @@ If Alpha and Dim are the same number, the strength used will be 1 and will have
                 },
                 optimizerType: {
                     type: 'string',
-                    description: `The optimizer determines how to update the neural net weights during training. 
-Various methods have been proposed for smart learning, but the most commonly used in LoRA learning 
+                    description: `The optimizer determines how to update the neural net weights during training.
+Various methods have been proposed for smart learning, but the most commonly used in LoRA learning
 is "AdamW8bit" or "Adafactor" for SDXL.`,
                     nullable: true,
                 },
@@ -2140,6 +2142,118 @@ export const $MochiVideoGenJob = {
         },
     },
 };
+export const $MusubiImageResourceTrainingInput = {
+    required: ['engine'],
+    allOf: [
+        {
+            $ref: '#/components/schemas/ImageResourceTrainingInput',
+        },
+        {
+            type: 'object',
+            properties: {
+                maxTrainEpochs: {
+                    maximum: 20,
+                    minimum: 0,
+                    type: 'integer',
+                    description: 'An epoch is one set of learning. By default, we will save a maximum of 20 epochs (evenly distributed), and they are all available for download.',
+                    format: 'int32',
+                    default: 5,
+                },
+                numRepeats: {
+                    maximum: 5000,
+                    minimum: 1,
+                    type: 'integer',
+                    description: 'Num Repeats defines how many times each individual image gets put into VRAM. As opposed to batch size, which is how many images are placed into VRAM at once.',
+                    format: 'int32',
+                    default: 8,
+                },
+                trainBatchSize: {
+                    maximum: 9,
+                    minimum: 1,
+                    type: 'integer',
+                    description: 'Batch size is the number of images that will be placed into VRAM at once. A batch size of 2 will train two images at a time, simultaneously.',
+                    format: 'int32',
+                    nullable: true,
+                },
+                resolution: {
+                    maximum: 1024,
+                    minimum: 512,
+                    type: 'integer',
+                    description: 'Specify the maximum resolution of training images. If the training images exceed the resolution specified here, they will be scaled down to this resolution',
+                    format: 'int32',
+                    nullable: true,
+                },
+                enableBucket: {
+                    type: 'boolean',
+                    description: 'Sorts images into buckets by size for the purposes of training. If your training images are all the same size, you can turn this option off, but leaving it on has no effect.',
+                    default: true,
+                },
+                unetLR: {
+                    maximum: 1,
+                    minimum: 0,
+                    type: 'number',
+                    description: 'Sets the learning rate for U-Net. This is the learning rate when performing additional learning on each attention block (and other blocks depending on the setting) in U-Net',
+                    format: 'double',
+                    default: 0.00005,
+                },
+                lrScheduler: {
+                    enum: ['constant', 'cosine', 'cosine_with_restarts', 'linear'],
+                    type: 'string',
+                    description: 'You can change the learning rate in the middle of learning. A scheduler is a setting for how to change the learning rate.',
+                    nullable: true,
+                },
+                lrSchedulerNumCycles: {
+                    maximum: 4,
+                    minimum: 1,
+                    type: 'integer',
+                    description: 'This option specifies how many cycles the scheduler runs during training. It is only used when "cosine_with_restarts" or "polynomial" is used as the scheduler.',
+                    format: 'int32',
+                    default: 3,
+                },
+                networkDim: {
+                    maximum: 256,
+                    minimum: 1,
+                    type: 'integer',
+                    description: 'The larger the Dim setting, the more learning information can be stored, but the possibility of learning unnecessary information other than the learning target increases. A larger Dim also increases LoRA file size.',
+                    format: 'int32',
+                    nullable: true,
+                },
+                networkAlpha: {
+                    maximum: 256,
+                    minimum: 1,
+                    type: 'integer',
+                    description: `The smaller the Network alpha value, the larger the stored LoRA neural net weights.
+For example, with an Alpha of 16 and a Dim of 32, the strength of the weight used is 16/32 = 0.5,
+meaning that the learning rate is only half as powerful as the Learning Rate setting.
+            
+If Alpha and Dim are the same number, the strength used will be 1 and will have no effect on the learning rate.`,
+                    format: 'int32',
+                    nullable: true,
+                },
+                optimizerType: {
+                    type: 'string',
+                    description: `The optimizer determines how to update the neural net weights during training.
+Various methods have been proposed for smart learning, but the most commonly used in LoRA learning
+is "AdamW8bit" or "Adafactor" for SDXL.`,
+                    nullable: true,
+                },
+                targetSteps: {
+                    type: 'integer',
+                    format: 'int32',
+                    nullable: true,
+                    readOnly: true,
+                },
+            },
+            additionalProperties: false,
+        },
+    ],
+    properties: {
+        engine: {
+            enum: ['musubi'],
+            type: 'string',
+        },
+    },
+};
 export const $NSFWLevel = {
     enum: ['pg', 'pG13', 'r', 'x', 'xxx', 'na'],
     type: 'string',
@@ -3345,6 +3459,55 @@ export const $ViduVideoGenStyle = {
     enum: ['general', 'anime'],
     type: 'string',
 };
+export const $WDTaggingJob = {
+    required: ['$type'],
+    allOf: [
+        {
+            $ref: '#/components/schemas/Job',
+        },
+        {
+            type: 'object',
+            properties: {
+                model: {
+                    type: 'string',
+                },
+                mediaUrl: {
+                    type: 'string',
+                    format: 'uri',
+                },
+                threshold: {
+                    type: 'number',
+                    format: 'double',
+                    nullable: true,
+                },
+                movieRatingModel: {
+                    type: 'string',
+                    nullable: true,
+                },
+                prompt: {
+                    type: 'string',
+                    nullable: true,
+                },
+                claimDuration: {
+                    type: 'string',
+                    format: 'date-span',
+                    readOnly: true,
+                },
+                type: {
+                    type: 'string',
+                    readOnly: true,
+                },
+            },
+            additionalProperties: false,
+        },
+    ],
+    properties: {
+        $type: {
+            enum: ['wdTagging'],
+            type: 'string',
+        },
+    },
+};
 export const $WorkerCapabilities = {
     type: 'object',
     properties: {
@@ -3800,6 +3963,12 @@ export const $WorkerRegistration = {
             description: 'Get additional metadata about this worker. This can be used for debugging purposes as well as to target certain configurations to certain workers.',
             nullable: true,
         },
+        capacitySaturationRate: {
+            type: 'number',
+            description: 'The saturation rate of available capacity of this worker of the worker.',
+            format: 'double',
+            nullable: true,
+        },
     },
     additionalProperties: false,
     description: "Details of a worker's registration.",
@@ -4113,6 +4282,12 @@ export const $WorkflowStep = {
             additionalProperties: {},
             description: 'A collection of user defined metadata for the workflow step.',
         },
+        estimatedProgressRate: {
+            type: 'number',
+            description: 'An estimation on the current progression of this step, or null if there is no estimation',
+            format: 'double',
+            nullable: true,
+        },
     },
     additionalProperties: false,
     description: 'Details of a workflow step.',
@@ -4183,6 +4358,12 @@ export const $WorkflowStepJob = {
             description: "The job's cost.",
             format: 'double',
         },
+        estimatedProgressRate: {
+            type: 'number',
+            description: 'An estimation on the current progression of this job, or null if there is no estimation',
+            format: 'double',
+            nullable: true,
+        },
     },
     additionalProperties: false,
     description: 'Details of a job produced by a workflow step.',

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

@@ -22,6 +22,7 @@ import type {
   GetRecommendedResourcesResponse,
   DownloadResourceData,
   CreateWorkerData,
+  CreateWorkerError,
   QueryWorkersResponse,
   GetWorkerData,
   DeleteWorkerData,
@@ -203,7 +204,7 @@ export declare const createWorker: (
   options?: Options<CreateWorkerData>
 ) => import('@hey-api/client-fetch').RequestResult<
   import('./types.gen').CreateWorkerResult,
-  import('./types.gen').ProblemDetails
+  CreateWorkerError
 >;
 /**
  * Query existing workers.

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

@@ -1108,6 +1108,65 @@ export type $type8 = 'mochi';
 export declare const $type8: {
   readonly MOCHI: 'mochi';
 };
+export type MusubiImageResourceTrainingInput = ImageResourceTrainingInput & {
+  /**
+   * An epoch is one set of learning. By default, we will save a maximum of 20 epochs (evenly distributed), and they are all available for download.
+   */
+  maxTrainEpochs?: number;
+  /**
+   * Num Repeats defines how many times each individual image gets put into VRAM. As opposed to batch size, which is how many images are placed into VRAM at once.
+   */
+  numRepeats?: number;
+  /**
+   * Batch size is the number of images that will be placed into VRAM at once. A batch size of 2 will train two images at a time, simultaneously.
+   */
+  trainBatchSize?: number | null;
+  /**
+   * Specify the maximum resolution of training images. If the training images exceed the resolution specified here, they will be scaled down to this resolution
+   */
+  resolution?: number | null;
+  /**
+   * Sorts images into buckets by size for the purposes of training. If your training images are all the same size, you can turn this option off, but leaving it on has no effect.
+   */
+  enableBucket?: boolean;
+  /**
+   * Sets the learning rate for U-Net. This is the learning rate when performing additional learning on each attention block (and other blocks depending on the setting) in U-Net
+   */
+  unetLR?: number;
+  /**
+   * You can change the learning rate in the middle of learning. A scheduler is a setting for how to change the learning rate.
+   */
+  lrScheduler?: 'constant' | 'cosine' | 'cosine_with_restarts' | 'linear' | null;
+  /**
+   * This option specifies how many cycles the scheduler runs during training. It is only used when "cosine_with_restarts" or "polynomial" is used as the scheduler.
+   */
+  lrSchedulerNumCycles?: number;
+  /**
+   * The larger the Dim setting, the more learning information can be stored, but the possibility of learning unnecessary information other than the learning target increases. A larger Dim also increases LoRA file size.
+   */
+  networkDim?: number | null;
+  /**
+   * The smaller the Network alpha value, the larger the stored LoRA neural net weights.
+   * For example, with an Alpha of 16 and a Dim of 32, the strength of the weight used is 16/32 = 0.5,
+   * meaning that the learning rate is only half as powerful as the Learning Rate setting.
+   *
+   * If Alpha and Dim are the same number, the strength used will be 1 and will have no effect on the learning rate.
+   */
+  networkAlpha?: number | null;
+  /**
+   * The optimizer determines how to update the neural net weights during training.
+   * Various methods have been proposed for smart learning, but the most commonly used in LoRA learning
+   * is "AdamW8bit" or "Adafactor" for SDXL.
+   */
+  optimizerType?: string | null;
+  readonly targetSteps?: number | null;
+} & {
+  engine: 'musubi';
+};
+export type engine9 = 'musubi';
+export declare const engine9: {
+  readonly MUSUBI: 'musubi';
+};
 export type NSFWLevel = 'pg' | 'pG13' | 'r' | 'x' | 'xxx' | 'na';
 export declare const NSFWLevel: {
   readonly PG: 'pg';
@@ -1783,8 +1842,8 @@ export declare const duration2: {
   readonly _4: 4;
   readonly _8: 8;
 };
-export type engine9 = 'vidu';
-export declare const engine9: {
+export type engine10 = 'vidu';
+export declare const engine10: {
   readonly VIDU: 'vidu';
 };
 export type ViduVideoGenJob = Job & {
@@ -1810,6 +1869,21 @@ export declare const ViduVideoGenStyle: {
   readonly GENERAL: 'general';
   readonly ANIME: 'anime';
 };
+export type WDTaggingJob = Job & {
+  model?: string;
+  mediaUrl?: string;
+  threshold?: number | null;
+  movieRatingModel?: string | null;
+  prompt?: string | null;
+  readonly claimDuration?: string;
+  readonly type?: string;
+} & {
+  $type: 'wdTagging';
+};
+export type $type16 = 'wdTagging';
+export declare const $type16: {
+  readonly WD_TAGGING: 'wdTagging';
+};
 /**
  * Details of a worker's capabilities.
  */
@@ -2123,6 +2197,10 @@ export type WorkerRegistration = {
   metadata?: {
     [key: string]: unknown;
   } | null;
+  /**
+   * The saturation rate of available capacity of this worker of the worker.
+   */
+  capacitySaturationRate?: number | null;
 };
 /**
  * Options for representing the status for a resource on a worker.
@@ -2379,6 +2457,10 @@ export type WorkflowStep = {
   metadata?: {
     [key: string]: unknown;
   };
+  /**
+   * An estimation on the current progression of this step, or null if there is no estimation
+   */
+  estimatedProgressRate?: number | null;
 };
 /**
  * Details of a workflow step event.
@@ -2417,6 +2499,10 @@ export type WorkflowStepJob = {
    * The job's cost.
    */
   cost?: number;
+  /**
+   * An estimation on the current progression of this job, or null if there is no estimation
+   */
+  estimatedProgressRate?: number | null;
 };
 /**
  * Details of a workflow step job event.
@@ -2693,7 +2779,7 @@ export type CreateWorkerData = {
   body?: WorkerRegistration;
 };
 export type CreateWorkerResponse = CreateWorkerResult;
-export type CreateWorkerError = ProblemDetails;
+export type CreateWorkerError = ProblemDetails & string;
 export type QueryWorkersResponse = Array<WorkerDetails>;
 export type QueryWorkersError = ProblemDetails;
 export type GetWorkerData = {
@@ -3366,6 +3452,10 @@ export type $OpenApiTs = {
          * Forbidden
          */
         '403': ProblemDetails;
+        /**
+         * Not Acceptable
+         */
+        '406': string;
       };
     };
     get: {

package/dist/generated/types.gen.js

@@ -187,6 +187,9 @@ export const engine8 = {
 export const $type8 = {
     MOCHI: 'mochi',
 };
+export const engine9 = {
+    MUSUBI: 'musubi',
+};
 export const NSFWLevel = {
     PG: 'pg',
     P_G13: 'pG13',
@@ -272,7 +275,7 @@ export const duration2 = {
     _4: 4,
     _8: 8,
 };
-export const engine9 = {
+export const engine10 = {
     VIDU: 'vidu',
 };
 export const $type15 = {
@@ -282,6 +285,9 @@ export const ViduVideoGenStyle = {
     GENERAL: 'general',
     ANIME: 'anime',
 };
+export const $type16 = {
+    WD_TAGGING: 'wdTagging',
+};
 /**
  * Options for representing the status for a resource on a worker.
  */

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@civitai/client",
-  "version": "0.1.9-beta.24",
+  "version": "0.1.9-beta.26",
   "description": "Civitai's javascript client for generating ai content",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",