@f-inverse/jammi-client 0.25.0 → 0.26.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +2 -2
- package/dist/gen/jammi/v1/audit_pb.d.ts +1 -1
- package/dist/gen/jammi/v1/audit_pb.js +1 -1
- package/dist/gen/jammi/v1/catalog_pb.d.ts +1148 -0
- package/dist/gen/jammi/v1/catalog_pb.d.ts.map +1 -0
- package/dist/gen/jammi/v1/catalog_pb.js +320 -0
- package/dist/gen/jammi/v1/catalog_pb.js.map +1 -0
- package/dist/gen/jammi/v1/embedding_pb.d.ts +10 -265
- package/dist/gen/jammi/v1/embedding_pb.d.ts.map +1 -1
- package/dist/gen/jammi/v1/embedding_pb.js +18 -111
- package/dist/gen/jammi/v1/embedding_pb.js.map +1 -1
- package/dist/gen/jammi/v1/error_pb.d.ts +2 -2
- package/dist/gen/jammi/v1/eval_pb.d.ts +287 -1
- package/dist/gen/jammi/v1/eval_pb.d.ts.map +1 -1
- package/dist/gen/jammi/v1/eval_pb.js +82 -25
- package/dist/gen/jammi/v1/eval_pb.js.map +1 -1
- package/dist/gen/jammi/v1/inference_pb.d.ts +272 -1
- package/dist/gen/jammi/v1/inference_pb.d.ts.map +1 -1
- package/dist/gen/jammi/v1/inference_pb.js +61 -2
- package/dist/gen/jammi/v1/inference_pb.js.map +1 -1
- package/dist/gen/jammi/v1/pipeline_pb.d.ts +491 -0
- package/dist/gen/jammi/v1/pipeline_pb.d.ts.map +1 -0
- package/dist/gen/jammi/v1/pipeline_pb.js +154 -0
- package/dist/gen/jammi/v1/pipeline_pb.js.map +1 -0
- package/dist/gen/jammi/v1/training_pb.d.ts +1217 -0
- package/dist/gen/jammi/v1/training_pb.d.ts.map +1 -0
- package/dist/gen/jammi/v1/training_pb.js +388 -0
- package/dist/gen/jammi/v1/training_pb.js.map +1 -0
- package/dist/gen/jammi/v1/trigger_pb.d.ts +18 -215
- package/dist/gen/jammi/v1/trigger_pb.d.ts.map +1 -1
- package/dist/gen/jammi/v1/trigger_pb.js +11 -41
- package/dist/gen/jammi/v1/trigger_pb.js.map +1 -1
- package/dist/index.d.ts +11 -12
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +10 -13
- package/dist/index.js.map +1 -1
- package/package.json +2 -2
- package/dist/gen/jammi/v1/channel_pb.d.ts +0 -158
- package/dist/gen/jammi/v1/channel_pb.d.ts.map +0 -1
- package/dist/gen/jammi/v1/channel_pb.js +0 -87
- package/dist/gen/jammi/v1/channel_pb.js.map +0 -1
- package/dist/gen/jammi/v1/fine_tune_pb.d.ts +0 -706
- package/dist/gen/jammi/v1/fine_tune_pb.d.ts.map +0 -1
- package/dist/gen/jammi/v1/fine_tune_pb.js +0 -271
- package/dist/gen/jammi/v1/fine_tune_pb.js.map +0 -1
- package/dist/gen/jammi/v1/mutable_table_pb.d.ts +0 -183
- package/dist/gen/jammi/v1/mutable_table_pb.d.ts.map +0 -1
- package/dist/gen/jammi/v1/mutable_table_pb.js +0 -59
- package/dist/gen/jammi/v1/mutable_table_pb.js.map +0 -1
- package/dist/gen/jammi/v1/session_pb.d.ts +0 -167
- package/dist/gen/jammi/v1/session_pb.d.ts.map +0 -1
- package/dist/gen/jammi/v1/session_pb.js +0 -45
- package/dist/gen/jammi/v1/session_pb.js.map +0 -1
|
@@ -1,706 +0,0 @@
|
|
|
1
|
-
import type { GenEnum, GenFile, GenMessage, GenService } from "@bufbuild/protobuf/codegenv2";
|
|
2
|
-
import type { ModelTask } from "./inference_pb";
|
|
3
|
-
import type { Message } from "@bufbuild/protobuf";
|
|
4
|
-
/**
|
|
5
|
-
* Describes the file jammi/v1/fine_tune.proto.
|
|
6
|
-
*/
|
|
7
|
-
export declare const file_jammi_v1_fine_tune: GenFile;
|
|
8
|
-
/**
|
|
9
|
-
* Embedding-loss function. Mirrors the engine's `EmbeddingLoss`; the two
|
|
10
|
-
* margin/temperature-carrying variants carry their scalar parameter.
|
|
11
|
-
*
|
|
12
|
-
* @generated from message jammi.v1.fine_tune.EmbeddingLoss
|
|
13
|
-
*/
|
|
14
|
-
export type EmbeddingLoss = Message<"jammi.v1.fine_tune.EmbeddingLoss"> & {
|
|
15
|
-
/**
|
|
16
|
-
* @generated from oneof jammi.v1.fine_tune.EmbeddingLoss.loss
|
|
17
|
-
*/
|
|
18
|
-
loss: {
|
|
19
|
-
/**
|
|
20
|
-
* CoSENT: cross-entropy over cosine-similarity ordering. No parameter.
|
|
21
|
-
*
|
|
22
|
-
* @generated from field: jammi.v1.fine_tune.EmbeddingLoss.CoSent co_sent = 1;
|
|
23
|
-
*/
|
|
24
|
-
value: EmbeddingLoss_CoSent;
|
|
25
|
-
case: "coSent";
|
|
26
|
-
} | {
|
|
27
|
-
/**
|
|
28
|
-
* Triplet loss with the given margin.
|
|
29
|
-
*
|
|
30
|
-
* @generated from field: jammi.v1.fine_tune.EmbeddingLoss.Triplet triplet = 2;
|
|
31
|
-
*/
|
|
32
|
-
value: EmbeddingLoss_Triplet;
|
|
33
|
-
case: "triplet";
|
|
34
|
-
} | {
|
|
35
|
-
/**
|
|
36
|
-
* InfoNCE with in-batch negatives at the given temperature.
|
|
37
|
-
*
|
|
38
|
-
* @generated from field: jammi.v1.fine_tune.EmbeddingLoss.MultipleNegativesRanking multiple_negatives_ranking = 3;
|
|
39
|
-
*/
|
|
40
|
-
value: EmbeddingLoss_MultipleNegativesRanking;
|
|
41
|
-
case: "multipleNegativesRanking";
|
|
42
|
-
} | {
|
|
43
|
-
/**
|
|
44
|
-
* AnglE: angle-difference objective in complex space. No parameter.
|
|
45
|
-
*
|
|
46
|
-
* @generated from field: jammi.v1.fine_tune.EmbeddingLoss.AnglE angle = 4;
|
|
47
|
-
*/
|
|
48
|
-
value: EmbeddingLoss_AnglE;
|
|
49
|
-
case: "angle";
|
|
50
|
-
} | {
|
|
51
|
-
/**
|
|
52
|
-
* cosine-MSE: regress scaled cosine onto a graded score. No parameter.
|
|
53
|
-
*
|
|
54
|
-
* @generated from field: jammi.v1.fine_tune.EmbeddingLoss.CosineMse cosine_mse = 5;
|
|
55
|
-
*/
|
|
56
|
-
value: EmbeddingLoss_CosineMse;
|
|
57
|
-
case: "cosineMse";
|
|
58
|
-
} | {
|
|
59
|
-
case: undefined;
|
|
60
|
-
value?: undefined;
|
|
61
|
-
};
|
|
62
|
-
};
|
|
63
|
-
/**
|
|
64
|
-
* Describes the message jammi.v1.fine_tune.EmbeddingLoss.
|
|
65
|
-
* Use `create(EmbeddingLossSchema)` to create a new message.
|
|
66
|
-
*/
|
|
67
|
-
export declare const EmbeddingLossSchema: GenMessage<EmbeddingLoss>;
|
|
68
|
-
/**
|
|
69
|
-
* @generated from message jammi.v1.fine_tune.EmbeddingLoss.CoSent
|
|
70
|
-
*/
|
|
71
|
-
export type EmbeddingLoss_CoSent = Message<"jammi.v1.fine_tune.EmbeddingLoss.CoSent"> & {};
|
|
72
|
-
/**
|
|
73
|
-
* Describes the message jammi.v1.fine_tune.EmbeddingLoss.CoSent.
|
|
74
|
-
* Use `create(EmbeddingLoss_CoSentSchema)` to create a new message.
|
|
75
|
-
*/
|
|
76
|
-
export declare const EmbeddingLoss_CoSentSchema: GenMessage<EmbeddingLoss_CoSent>;
|
|
77
|
-
/**
|
|
78
|
-
* @generated from message jammi.v1.fine_tune.EmbeddingLoss.Triplet
|
|
79
|
-
*/
|
|
80
|
-
export type EmbeddingLoss_Triplet = Message<"jammi.v1.fine_tune.EmbeddingLoss.Triplet"> & {
|
|
81
|
-
/**
|
|
82
|
-
* @generated from field: double margin = 1;
|
|
83
|
-
*/
|
|
84
|
-
margin: number;
|
|
85
|
-
};
|
|
86
|
-
/**
|
|
87
|
-
* Describes the message jammi.v1.fine_tune.EmbeddingLoss.Triplet.
|
|
88
|
-
* Use `create(EmbeddingLoss_TripletSchema)` to create a new message.
|
|
89
|
-
*/
|
|
90
|
-
export declare const EmbeddingLoss_TripletSchema: GenMessage<EmbeddingLoss_Triplet>;
|
|
91
|
-
/**
|
|
92
|
-
* @generated from message jammi.v1.fine_tune.EmbeddingLoss.MultipleNegativesRanking
|
|
93
|
-
*/
|
|
94
|
-
export type EmbeddingLoss_MultipleNegativesRanking = Message<"jammi.v1.fine_tune.EmbeddingLoss.MultipleNegativesRanking"> & {
|
|
95
|
-
/**
|
|
96
|
-
* @generated from field: double temperature = 1;
|
|
97
|
-
*/
|
|
98
|
-
temperature: number;
|
|
99
|
-
};
|
|
100
|
-
/**
|
|
101
|
-
* Describes the message jammi.v1.fine_tune.EmbeddingLoss.MultipleNegativesRanking.
|
|
102
|
-
* Use `create(EmbeddingLoss_MultipleNegativesRankingSchema)` to create a new message.
|
|
103
|
-
*/
|
|
104
|
-
export declare const EmbeddingLoss_MultipleNegativesRankingSchema: GenMessage<EmbeddingLoss_MultipleNegativesRanking>;
|
|
105
|
-
/**
|
|
106
|
-
* @generated from message jammi.v1.fine_tune.EmbeddingLoss.AnglE
|
|
107
|
-
*/
|
|
108
|
-
export type EmbeddingLoss_AnglE = Message<"jammi.v1.fine_tune.EmbeddingLoss.AnglE"> & {};
|
|
109
|
-
/**
|
|
110
|
-
* Describes the message jammi.v1.fine_tune.EmbeddingLoss.AnglE.
|
|
111
|
-
* Use `create(EmbeddingLoss_AnglESchema)` to create a new message.
|
|
112
|
-
*/
|
|
113
|
-
export declare const EmbeddingLoss_AnglESchema: GenMessage<EmbeddingLoss_AnglE>;
|
|
114
|
-
/**
|
|
115
|
-
* @generated from message jammi.v1.fine_tune.EmbeddingLoss.CosineMse
|
|
116
|
-
*/
|
|
117
|
-
export type EmbeddingLoss_CosineMse = Message<"jammi.v1.fine_tune.EmbeddingLoss.CosineMse"> & {};
|
|
118
|
-
/**
|
|
119
|
-
* Describes the message jammi.v1.fine_tune.EmbeddingLoss.CosineMse.
|
|
120
|
-
* Use `create(EmbeddingLoss_CosineMseSchema)` to create a new message.
|
|
121
|
-
*/
|
|
122
|
-
export declare const EmbeddingLoss_CosineMseSchema: GenMessage<EmbeddingLoss_CosineMse>;
|
|
123
|
-
/**
|
|
124
|
-
* Proper-scoring objective for a distributional regression head (S18). Mirrors
|
|
125
|
-
* the engine's `RegressionLoss`; the β-NLL variant carries its weighting
|
|
126
|
-
* exponent. Pinball trains the quantile head over `quantile_levels`; the others
|
|
127
|
-
* train the parametric Gaussian head.
|
|
128
|
-
*
|
|
129
|
-
* @generated from message jammi.v1.fine_tune.RegressionLoss
|
|
130
|
-
*/
|
|
131
|
-
export type RegressionLoss = Message<"jammi.v1.fine_tune.RegressionLoss"> & {
|
|
132
|
-
/**
|
|
133
|
-
* @generated from oneof jammi.v1.fine_tune.RegressionLoss.loss
|
|
134
|
-
*/
|
|
135
|
-
loss: {
|
|
136
|
-
/**
|
|
137
|
-
* Gaussian NLL — the plain heteroscedastic mean-variance objective.
|
|
138
|
-
*
|
|
139
|
-
* @generated from field: jammi.v1.fine_tune.RegressionLoss.GaussianNll gaussian_nll = 1;
|
|
140
|
-
*/
|
|
141
|
-
value: RegressionLoss_GaussianNll;
|
|
142
|
-
case: "gaussianNll";
|
|
143
|
-
} | {
|
|
144
|
-
/**
|
|
145
|
-
* β-NLL with the given variance-weighting exponent (Seitzer 2022).
|
|
146
|
-
*
|
|
147
|
-
* @generated from field: jammi.v1.fine_tune.RegressionLoss.BetaNll beta_nll = 2;
|
|
148
|
-
*/
|
|
149
|
-
value: RegressionLoss_BetaNll;
|
|
150
|
-
case: "betaNll";
|
|
151
|
-
} | {
|
|
152
|
-
/**
|
|
153
|
-
* Closed-form Gaussian CRPS. No parameter.
|
|
154
|
-
*
|
|
155
|
-
* @generated from field: jammi.v1.fine_tune.RegressionLoss.Crps crps = 3;
|
|
156
|
-
*/
|
|
157
|
-
value: RegressionLoss_Crps;
|
|
158
|
-
case: "crps";
|
|
159
|
-
} | {
|
|
160
|
-
/**
|
|
161
|
-
* Pinball / quantile loss over the configured levels. No parameter.
|
|
162
|
-
*
|
|
163
|
-
* @generated from field: jammi.v1.fine_tune.RegressionLoss.Pinball pinball = 4;
|
|
164
|
-
*/
|
|
165
|
-
value: RegressionLoss_Pinball;
|
|
166
|
-
case: "pinball";
|
|
167
|
-
} | {
|
|
168
|
-
case: undefined;
|
|
169
|
-
value?: undefined;
|
|
170
|
-
};
|
|
171
|
-
};
|
|
172
|
-
/**
|
|
173
|
-
* Describes the message jammi.v1.fine_tune.RegressionLoss.
|
|
174
|
-
* Use `create(RegressionLossSchema)` to create a new message.
|
|
175
|
-
*/
|
|
176
|
-
export declare const RegressionLossSchema: GenMessage<RegressionLoss>;
|
|
177
|
-
/**
|
|
178
|
-
* @generated from message jammi.v1.fine_tune.RegressionLoss.GaussianNll
|
|
179
|
-
*/
|
|
180
|
-
export type RegressionLoss_GaussianNll = Message<"jammi.v1.fine_tune.RegressionLoss.GaussianNll"> & {};
|
|
181
|
-
/**
|
|
182
|
-
* Describes the message jammi.v1.fine_tune.RegressionLoss.GaussianNll.
|
|
183
|
-
* Use `create(RegressionLoss_GaussianNllSchema)` to create a new message.
|
|
184
|
-
*/
|
|
185
|
-
export declare const RegressionLoss_GaussianNllSchema: GenMessage<RegressionLoss_GaussianNll>;
|
|
186
|
-
/**
|
|
187
|
-
* @generated from message jammi.v1.fine_tune.RegressionLoss.BetaNll
|
|
188
|
-
*/
|
|
189
|
-
export type RegressionLoss_BetaNll = Message<"jammi.v1.fine_tune.RegressionLoss.BetaNll"> & {
|
|
190
|
-
/**
|
|
191
|
-
* @generated from field: double beta = 1;
|
|
192
|
-
*/
|
|
193
|
-
beta: number;
|
|
194
|
-
};
|
|
195
|
-
/**
|
|
196
|
-
* Describes the message jammi.v1.fine_tune.RegressionLoss.BetaNll.
|
|
197
|
-
* Use `create(RegressionLoss_BetaNllSchema)` to create a new message.
|
|
198
|
-
*/
|
|
199
|
-
export declare const RegressionLoss_BetaNllSchema: GenMessage<RegressionLoss_BetaNll>;
|
|
200
|
-
/**
|
|
201
|
-
* @generated from message jammi.v1.fine_tune.RegressionLoss.Crps
|
|
202
|
-
*/
|
|
203
|
-
export type RegressionLoss_Crps = Message<"jammi.v1.fine_tune.RegressionLoss.Crps"> & {};
|
|
204
|
-
/**
|
|
205
|
-
* Describes the message jammi.v1.fine_tune.RegressionLoss.Crps.
|
|
206
|
-
* Use `create(RegressionLoss_CrpsSchema)` to create a new message.
|
|
207
|
-
*/
|
|
208
|
-
export declare const RegressionLoss_CrpsSchema: GenMessage<RegressionLoss_Crps>;
|
|
209
|
-
/**
|
|
210
|
-
* @generated from message jammi.v1.fine_tune.RegressionLoss.Pinball
|
|
211
|
-
*/
|
|
212
|
-
export type RegressionLoss_Pinball = Message<"jammi.v1.fine_tune.RegressionLoss.Pinball"> & {};
|
|
213
|
-
/**
|
|
214
|
-
* Describes the message jammi.v1.fine_tune.RegressionLoss.Pinball.
|
|
215
|
-
* Use `create(RegressionLoss_PinballSchema)` to create a new message.
|
|
216
|
-
*/
|
|
217
|
-
export declare const RegressionLoss_PinballSchema: GenMessage<RegressionLoss_Pinball>;
|
|
218
|
-
/**
|
|
219
|
-
* 0-based encoder layer indices that receive LoRA adapters. Wrapper message so
|
|
220
|
-
* "no restriction" (field absent → all layers) is distinguishable from "an
|
|
221
|
-
* empty restriction"; mirrors the engine's `Option<Vec<usize>>`.
|
|
222
|
-
*
|
|
223
|
-
* @generated from message jammi.v1.fine_tune.LayersToTransform
|
|
224
|
-
*/
|
|
225
|
-
export type LayersToTransform = Message<"jammi.v1.fine_tune.LayersToTransform"> & {
|
|
226
|
-
/**
|
|
227
|
-
* @generated from field: repeated uint32 layers = 1;
|
|
228
|
-
*/
|
|
229
|
-
layers: number[];
|
|
230
|
-
};
|
|
231
|
-
/**
|
|
232
|
-
* Describes the message jammi.v1.fine_tune.LayersToTransform.
|
|
233
|
-
* Use `create(LayersToTransformSchema)` to create a new message.
|
|
234
|
-
*/
|
|
235
|
-
export declare const LayersToTransformSchema: GenMessage<LayersToTransform>;
|
|
236
|
-
/**
|
|
237
|
-
* Hard-negative mining configuration. Mirrors the engine's
|
|
238
|
-
* `HardNegativeConfig`. When `mine` is set the trainer mines hard negatives
|
|
239
|
-
* from its own ANN index, excluding the positive's `exclude_hops`-hop
|
|
240
|
-
* neighbourhood and re-mining every `refresh_every` epochs.
|
|
241
|
-
*
|
|
242
|
-
* @generated from message jammi.v1.fine_tune.HardNegativeConfig
|
|
243
|
-
*/
|
|
244
|
-
export type HardNegativeConfig = Message<"jammi.v1.fine_tune.HardNegativeConfig"> & {
|
|
245
|
-
/**
|
|
246
|
-
* @generated from field: bool mine = 1;
|
|
247
|
-
*/
|
|
248
|
-
mine: boolean;
|
|
249
|
-
/**
|
|
250
|
-
* @generated from field: uint32 k = 2;
|
|
251
|
-
*/
|
|
252
|
-
k: number;
|
|
253
|
-
/**
|
|
254
|
-
* @generated from field: uint32 exclude_hops = 3;
|
|
255
|
-
*/
|
|
256
|
-
excludeHops: number;
|
|
257
|
-
/**
|
|
258
|
-
* @generated from field: uint32 refresh_every = 4;
|
|
259
|
-
*/
|
|
260
|
-
refreshEvery: number;
|
|
261
|
-
};
|
|
262
|
-
/**
|
|
263
|
-
* Describes the message jammi.v1.fine_tune.HardNegativeConfig.
|
|
264
|
-
* Use `create(HardNegativeConfigSchema)` to create a new message.
|
|
265
|
-
*/
|
|
266
|
-
export declare const HardNegativeConfigSchema: GenMessage<HardNegativeConfig>;
|
|
267
|
-
/**
|
|
268
|
-
* Configuration for a fine-tuning job. Mirrors the engine's `FineTuneConfig`
|
|
269
|
-
* field for field; every field is read by the trainer or the LoRA builder.
|
|
270
|
-
* `embedding_loss`/`classification_loss` are unset to let the engine
|
|
271
|
-
* auto-select from the data format; `layers_to_transform` is unset for "all
|
|
272
|
-
* layers". The enum-typed fields default to the engine's default variant when
|
|
273
|
-
* left UNSPECIFIED, matching `FineTuneConfig::default()`.
|
|
274
|
-
*
|
|
275
|
-
* @generated from message jammi.v1.fine_tune.FineTuneConfig
|
|
276
|
-
*/
|
|
277
|
-
export type FineTuneConfig = Message<"jammi.v1.fine_tune.FineTuneConfig"> & {
|
|
278
|
-
/**
|
|
279
|
-
* @generated from field: uint32 lora_rank = 1;
|
|
280
|
-
*/
|
|
281
|
-
loraRank: number;
|
|
282
|
-
/**
|
|
283
|
-
* @generated from field: double lora_alpha = 2;
|
|
284
|
-
*/
|
|
285
|
-
loraAlpha: number;
|
|
286
|
-
/**
|
|
287
|
-
* @generated from field: double lora_dropout = 3;
|
|
288
|
-
*/
|
|
289
|
-
loraDropout: number;
|
|
290
|
-
/**
|
|
291
|
-
* @generated from field: double learning_rate = 4;
|
|
292
|
-
*/
|
|
293
|
-
learningRate: number;
|
|
294
|
-
/**
|
|
295
|
-
* @generated from field: uint32 epochs = 5;
|
|
296
|
-
*/
|
|
297
|
-
epochs: number;
|
|
298
|
-
/**
|
|
299
|
-
* @generated from field: uint32 batch_size = 6;
|
|
300
|
-
*/
|
|
301
|
-
batchSize: number;
|
|
302
|
-
/**
|
|
303
|
-
* @generated from field: uint32 max_seq_length = 7;
|
|
304
|
-
*/
|
|
305
|
-
maxSeqLength: number;
|
|
306
|
-
/**
|
|
307
|
-
* Unset = auto-select from the data format.
|
|
308
|
-
*
|
|
309
|
-
* @generated from field: optional jammi.v1.fine_tune.EmbeddingLoss embedding_loss = 8;
|
|
310
|
-
*/
|
|
311
|
-
embeddingLoss?: EmbeddingLoss | undefined;
|
|
312
|
-
/**
|
|
313
|
-
* Unset = auto-select from the data format.
|
|
314
|
-
*
|
|
315
|
-
* @generated from field: optional jammi.v1.fine_tune.ClassificationLoss classification_loss = 9;
|
|
316
|
-
*/
|
|
317
|
-
classificationLoss?: ClassificationLoss | undefined;
|
|
318
|
-
/**
|
|
319
|
-
* @generated from field: uint32 gradient_accumulation_steps = 10;
|
|
320
|
-
*/
|
|
321
|
-
gradientAccumulationSteps: number;
|
|
322
|
-
/**
|
|
323
|
-
* @generated from field: double validation_fraction = 11;
|
|
324
|
-
*/
|
|
325
|
-
validationFraction: number;
|
|
326
|
-
/**
|
|
327
|
-
* @generated from field: uint32 early_stopping_patience = 12;
|
|
328
|
-
*/
|
|
329
|
-
earlyStoppingPatience: number;
|
|
330
|
-
/**
|
|
331
|
-
* @generated from field: uint32 warmup_steps = 13;
|
|
332
|
-
*/
|
|
333
|
-
warmupSteps: number;
|
|
334
|
-
/**
|
|
335
|
-
* @generated from field: jammi.v1.fine_tune.LrSchedule lr_schedule = 14;
|
|
336
|
-
*/
|
|
337
|
-
lrSchedule: LrSchedule;
|
|
338
|
-
/**
|
|
339
|
-
* @generated from field: jammi.v1.fine_tune.EarlyStoppingMetric early_stopping_metric = 15;
|
|
340
|
-
*/
|
|
341
|
-
earlyStoppingMetric: EarlyStoppingMetric;
|
|
342
|
-
/**
|
|
343
|
-
* Layer-name suffixes that receive LoRA adapters. Empty = projection head
|
|
344
|
-
* over the frozen base model; non-empty = LoRA inside the encoder.
|
|
345
|
-
*
|
|
346
|
-
* @generated from field: repeated string target_modules = 16;
|
|
347
|
-
*/
|
|
348
|
-
targetModules: string[];
|
|
349
|
-
/**
|
|
350
|
-
* Unset = all layers.
|
|
351
|
-
*
|
|
352
|
-
* @generated from field: optional jammi.v1.fine_tune.LayersToTransform layers_to_transform = 17;
|
|
353
|
-
*/
|
|
354
|
-
layersToTransform?: LayersToTransform | undefined;
|
|
355
|
-
/**
|
|
356
|
-
* @generated from field: bool use_rslora = 18;
|
|
357
|
-
*/
|
|
358
|
-
useRslora: boolean;
|
|
359
|
-
/**
|
|
360
|
-
* Per-module rank overrides keyed by module-name substring.
|
|
361
|
-
*
|
|
362
|
-
* @generated from field: map<string, uint32> rank_pattern = 19;
|
|
363
|
-
*/
|
|
364
|
-
rankPattern: {
|
|
365
|
-
[key: string]: number;
|
|
366
|
-
};
|
|
367
|
-
/**
|
|
368
|
-
* @generated from field: jammi.v1.fine_tune.LoraInitMode init_lora_weights = 20;
|
|
369
|
-
*/
|
|
370
|
-
initLoraWeights: LoraInitMode;
|
|
371
|
-
/**
|
|
372
|
-
* @generated from field: jammi.v1.fine_tune.BackboneDtype backbone_dtype = 21;
|
|
373
|
-
*/
|
|
374
|
-
backboneDtype: BackboneDtype;
|
|
375
|
-
/**
|
|
376
|
-
* @generated from field: double weight_decay = 22;
|
|
377
|
-
*/
|
|
378
|
-
weightDecay: number;
|
|
379
|
-
/**
|
|
380
|
-
* @generated from field: double max_grad_norm = 23;
|
|
381
|
-
*/
|
|
382
|
-
maxGradNorm: number;
|
|
383
|
-
/**
|
|
384
|
-
* GradCache: two-pass in-batch-negative training that enlarges the effective
|
|
385
|
-
* negative pool without the memory cost. Only affects MultipleNegativesRanking.
|
|
386
|
-
*
|
|
387
|
-
* @generated from field: bool cached = 24;
|
|
388
|
-
*/
|
|
389
|
-
cached: boolean;
|
|
390
|
-
/**
|
|
391
|
-
* Hard-negative mining over jammi's own ANN index. Field absent = mining off.
|
|
392
|
-
*
|
|
393
|
-
* @generated from field: optional jammi.v1.fine_tune.HardNegativeConfig hard_negatives = 25;
|
|
394
|
-
*/
|
|
395
|
-
hardNegatives?: HardNegativeConfig | undefined;
|
|
396
|
-
/**
|
|
397
|
-
* Matryoshka prefix dimensions. Empty = full dimension only.
|
|
398
|
-
*
|
|
399
|
-
* @generated from field: repeated uint32 matryoshka_dims = 26;
|
|
400
|
-
*/
|
|
401
|
-
matryoshkaDims: number[];
|
|
402
|
-
/**
|
|
403
|
-
* Unset = the engine default (β-NLL, β=0.5). The Pinball variant trains the
|
|
404
|
-
* quantile head over `quantile_levels`.
|
|
405
|
-
*
|
|
406
|
-
* @generated from field: optional jammi.v1.fine_tune.RegressionLoss regression_loss = 27;
|
|
407
|
-
*/
|
|
408
|
-
regressionLoss?: RegressionLoss | undefined;
|
|
409
|
-
/**
|
|
410
|
-
* Quantile levels for a pinball-trained regression head, ascending in (0, 1).
|
|
411
|
-
* Empty for the parametric Gaussian objectives.
|
|
412
|
-
*
|
|
413
|
-
* @generated from field: repeated double quantile_levels = 28;
|
|
414
|
-
*/
|
|
415
|
-
quantileLevels: number[];
|
|
416
|
-
};
|
|
417
|
-
/**
|
|
418
|
-
* Describes the message jammi.v1.fine_tune.FineTuneConfig.
|
|
419
|
-
* Use `create(FineTuneConfigSchema)` to create a new message.
|
|
420
|
-
*/
|
|
421
|
-
export declare const FineTuneConfigSchema: GenMessage<FineTuneConfig>;
|
|
422
|
-
/**
|
|
423
|
-
* @generated from message jammi.v1.fine_tune.StartFineTuneRequest
|
|
424
|
-
*/
|
|
425
|
-
export type StartFineTuneRequest = Message<"jammi.v1.fine_tune.StartFineTuneRequest"> & {
|
|
426
|
-
/**
|
|
427
|
-
* Registered source carrying the training data.
|
|
428
|
-
*
|
|
429
|
-
* @generated from field: string source_id = 1;
|
|
430
|
-
*/
|
|
431
|
-
sourceId: string;
|
|
432
|
-
/**
|
|
433
|
-
* Base encoder to adapt. `local:<path>`, an HF repo id, or a fine-tuned id.
|
|
434
|
-
*
|
|
435
|
-
* @generated from field: string base_model = 2;
|
|
436
|
-
*/
|
|
437
|
-
baseModel: string;
|
|
438
|
-
/**
|
|
439
|
-
* Training-data columns read from the source (e.g. text_a/text_b/score,
|
|
440
|
-
* anchor/positive/negative, or text/label — the engine detects the format).
|
|
441
|
-
*
|
|
442
|
-
* @generated from field: repeated string columns = 3;
|
|
443
|
-
*/
|
|
444
|
-
columns: string[];
|
|
445
|
-
/**
|
|
446
|
-
* Fine-tuning method.
|
|
447
|
-
*
|
|
448
|
-
* @generated from field: jammi.v1.fine_tune.FineTuneMethod method = 4;
|
|
449
|
-
*/
|
|
450
|
-
method: FineTuneMethod;
|
|
451
|
-
/**
|
|
452
|
-
* Task the adapted model performs. Reuses InferenceService's ModelTask.
|
|
453
|
-
*
|
|
454
|
-
* @generated from field: jammi.v1.inference.ModelTask task = 5;
|
|
455
|
-
*/
|
|
456
|
-
task: ModelTask;
|
|
457
|
-
/**
|
|
458
|
-
* Optional hyperparameters; unset applies the engine's defaults.
|
|
459
|
-
*
|
|
460
|
-
* @generated from field: optional jammi.v1.fine_tune.FineTuneConfig config = 6;
|
|
461
|
-
*/
|
|
462
|
-
config?: FineTuneConfig | undefined;
|
|
463
|
-
};
|
|
464
|
-
/**
|
|
465
|
-
* Describes the message jammi.v1.fine_tune.StartFineTuneRequest.
|
|
466
|
-
* Use `create(StartFineTuneRequestSchema)` to create a new message.
|
|
467
|
-
*/
|
|
468
|
-
export declare const StartFineTuneRequestSchema: GenMessage<StartFineTuneRequest>;
|
|
469
|
-
/**
|
|
470
|
-
* @generated from message jammi.v1.fine_tune.StartFineTuneResponse
|
|
471
|
-
*/
|
|
472
|
-
export type StartFineTuneResponse = Message<"jammi.v1.fine_tune.StartFineTuneResponse"> & {
|
|
473
|
-
/**
|
|
474
|
-
* Server-assigned job id. Poll `FineTuneStatus` with it.
|
|
475
|
-
*
|
|
476
|
-
* @generated from field: string job_id = 1;
|
|
477
|
-
*/
|
|
478
|
-
jobId: string;
|
|
479
|
-
};
|
|
480
|
-
/**
|
|
481
|
-
* Describes the message jammi.v1.fine_tune.StartFineTuneResponse.
|
|
482
|
-
* Use `create(StartFineTuneResponseSchema)` to create a new message.
|
|
483
|
-
*/
|
|
484
|
-
export declare const StartFineTuneResponseSchema: GenMessage<StartFineTuneResponse>;
|
|
485
|
-
/**
|
|
486
|
-
* @generated from message jammi.v1.fine_tune.FineTuneStatusRequest
|
|
487
|
-
*/
|
|
488
|
-
export type FineTuneStatusRequest = Message<"jammi.v1.fine_tune.FineTuneStatusRequest"> & {
|
|
489
|
-
/**
|
|
490
|
-
* Job id returned by a prior StartFineTune call.
|
|
491
|
-
*
|
|
492
|
-
* @generated from field: string job_id = 1;
|
|
493
|
-
*/
|
|
494
|
-
jobId: string;
|
|
495
|
-
};
|
|
496
|
-
/**
|
|
497
|
-
* Describes the message jammi.v1.fine_tune.FineTuneStatusRequest.
|
|
498
|
-
* Use `create(FineTuneStatusRequestSchema)` to create a new message.
|
|
499
|
-
*/
|
|
500
|
-
export declare const FineTuneStatusRequestSchema: GenMessage<FineTuneStatusRequest>;
|
|
501
|
-
/**
|
|
502
|
-
* @generated from message jammi.v1.fine_tune.FineTuneStatusResponse
|
|
503
|
-
*/
|
|
504
|
-
export type FineTuneStatusResponse = Message<"jammi.v1.fine_tune.FineTuneStatusResponse"> & {
|
|
505
|
-
/**
|
|
506
|
-
* Current lifecycle status: "queued", "running", "completed", or "failed".
|
|
507
|
-
*
|
|
508
|
-
* @generated from field: string status = 1;
|
|
509
|
-
*/
|
|
510
|
-
status: string;
|
|
511
|
-
};
|
|
512
|
-
/**
|
|
513
|
-
* Describes the message jammi.v1.fine_tune.FineTuneStatusResponse.
|
|
514
|
-
* Use `create(FineTuneStatusResponseSchema)` to create a new message.
|
|
515
|
-
*/
|
|
516
|
-
export declare const FineTuneStatusResponseSchema: GenMessage<FineTuneStatusResponse>;
|
|
517
|
-
/**
|
|
518
|
-
* Fine-tuning method. Mirrors the engine's `FineTuneMethod`; LoRA is the only
|
|
519
|
-
* method, trained as either an encoder-internal adapter or a projection head
|
|
520
|
-
* over a frozen base model depending on `FineTuneConfig.target_modules`.
|
|
521
|
-
*
|
|
522
|
-
* @generated from enum jammi.v1.fine_tune.FineTuneMethod
|
|
523
|
-
*/
|
|
524
|
-
export declare enum FineTuneMethod {
|
|
525
|
-
/**
|
|
526
|
-
* @generated from enum value: FINE_TUNE_METHOD_UNSPECIFIED = 0;
|
|
527
|
-
*/
|
|
528
|
-
FINE_TUNE_METHOD_UNSPECIFIED = 0,
|
|
529
|
-
/**
|
|
530
|
-
* @generated from enum value: LORA = 1;
|
|
531
|
-
*/
|
|
532
|
-
LORA = 1
|
|
533
|
-
}
|
|
534
|
-
/**
|
|
535
|
-
* Describes the enum jammi.v1.fine_tune.FineTuneMethod.
|
|
536
|
-
*/
|
|
537
|
-
export declare const FineTuneMethodSchema: GenEnum<FineTuneMethod>;
|
|
538
|
-
/**
|
|
539
|
-
* Classification-loss function. Mirrors the engine's `ClassificationLoss`.
|
|
540
|
-
*
|
|
541
|
-
* @generated from enum jammi.v1.fine_tune.ClassificationLoss
|
|
542
|
-
*/
|
|
543
|
-
export declare enum ClassificationLoss {
|
|
544
|
-
/**
|
|
545
|
-
* @generated from enum value: CLASSIFICATION_LOSS_UNSPECIFIED = 0;
|
|
546
|
-
*/
|
|
547
|
-
CLASSIFICATION_LOSS_UNSPECIFIED = 0,
|
|
548
|
-
/**
|
|
549
|
-
* @generated from enum value: CROSS_ENTROPY = 1;
|
|
550
|
-
*/
|
|
551
|
-
CROSS_ENTROPY = 1
|
|
552
|
-
}
|
|
553
|
-
/**
|
|
554
|
-
* Describes the enum jammi.v1.fine_tune.ClassificationLoss.
|
|
555
|
-
*/
|
|
556
|
-
export declare const ClassificationLossSchema: GenEnum<ClassificationLoss>;
|
|
557
|
-
/**
|
|
558
|
-
* Which loss signal drives early stopping. Mirrors `EarlyStoppingMetric`.
|
|
559
|
-
*
|
|
560
|
-
* @generated from enum jammi.v1.fine_tune.EarlyStoppingMetric
|
|
561
|
-
*/
|
|
562
|
-
export declare enum EarlyStoppingMetric {
|
|
563
|
-
/**
|
|
564
|
-
* @generated from enum value: EARLY_STOPPING_METRIC_UNSPECIFIED = 0;
|
|
565
|
-
*/
|
|
566
|
-
EARLY_STOPPING_METRIC_UNSPECIFIED = 0,
|
|
567
|
-
/**
|
|
568
|
-
* @generated from enum value: VAL_LOSS = 1;
|
|
569
|
-
*/
|
|
570
|
-
VAL_LOSS = 1,
|
|
571
|
-
/**
|
|
572
|
-
* @generated from enum value: TRAIN_LOSS = 2;
|
|
573
|
-
*/
|
|
574
|
-
TRAIN_LOSS = 2
|
|
575
|
-
}
|
|
576
|
-
/**
|
|
577
|
-
* Describes the enum jammi.v1.fine_tune.EarlyStoppingMetric.
|
|
578
|
-
*/
|
|
579
|
-
export declare const EarlyStoppingMetricSchema: GenEnum<EarlyStoppingMetric>;
|
|
580
|
-
/**
|
|
581
|
-
* Learning-rate schedule after warmup. Mirrors the engine's `LrSchedule`.
|
|
582
|
-
*
|
|
583
|
-
* @generated from enum jammi.v1.fine_tune.LrSchedule
|
|
584
|
-
*/
|
|
585
|
-
export declare enum LrSchedule {
|
|
586
|
-
/**
|
|
587
|
-
* @generated from enum value: LR_SCHEDULE_UNSPECIFIED = 0;
|
|
588
|
-
*/
|
|
589
|
-
LR_SCHEDULE_UNSPECIFIED = 0,
|
|
590
|
-
/**
|
|
591
|
-
* @generated from enum value: CONSTANT = 1;
|
|
592
|
-
*/
|
|
593
|
-
CONSTANT = 1,
|
|
594
|
-
/**
|
|
595
|
-
* @generated from enum value: COSINE_DECAY = 2;
|
|
596
|
-
*/
|
|
597
|
-
COSINE_DECAY = 2,
|
|
598
|
-
/**
|
|
599
|
-
* @generated from enum value: LINEAR_DECAY = 3;
|
|
600
|
-
*/
|
|
601
|
-
LINEAR_DECAY = 3
|
|
602
|
-
}
|
|
603
|
-
/**
|
|
604
|
-
* Describes the enum jammi.v1.fine_tune.LrSchedule.
|
|
605
|
-
*/
|
|
606
|
-
export declare const LrScheduleSchema: GenEnum<LrSchedule>;
|
|
607
|
-
/**
|
|
608
|
-
* LoRA A/B matrix initialisation. Mirrors `jammi_lora::LoraInitMode`.
|
|
609
|
-
*
|
|
610
|
-
* @generated from enum jammi.v1.fine_tune.LoraInitMode
|
|
611
|
-
*/
|
|
612
|
-
export declare enum LoraInitMode {
|
|
613
|
-
/**
|
|
614
|
-
* @generated from enum value: LORA_INIT_MODE_UNSPECIFIED = 0;
|
|
615
|
-
*/
|
|
616
|
-
LORA_INIT_MODE_UNSPECIFIED = 0,
|
|
617
|
-
/**
|
|
618
|
-
* @generated from enum value: ZEROS_B = 1;
|
|
619
|
-
*/
|
|
620
|
-
ZEROS_B = 1,
|
|
621
|
-
/**
|
|
622
|
-
* @generated from enum value: GAUSSIAN = 2;
|
|
623
|
-
*/
|
|
624
|
-
GAUSSIAN = 2
|
|
625
|
-
}
|
|
626
|
-
/**
|
|
627
|
-
* Describes the enum jammi.v1.fine_tune.LoraInitMode.
|
|
628
|
-
*/
|
|
629
|
-
export declare const LoraInitModeSchema: GenEnum<LoraInitMode>;
|
|
630
|
-
/**
|
|
631
|
-
* Dtype for the frozen backbone weights. Mirrors `jammi_lora::BackboneDtype`.
|
|
632
|
-
*
|
|
633
|
-
* @generated from enum jammi.v1.fine_tune.BackboneDtype
|
|
634
|
-
*/
|
|
635
|
-
export declare enum BackboneDtype {
|
|
636
|
-
/**
|
|
637
|
-
* @generated from enum value: BACKBONE_DTYPE_UNSPECIFIED = 0;
|
|
638
|
-
*/
|
|
639
|
-
BACKBONE_DTYPE_UNSPECIFIED = 0,
|
|
640
|
-
/**
|
|
641
|
-
* @generated from enum value: F32 = 1;
|
|
642
|
-
*/
|
|
643
|
-
F32 = 1,
|
|
644
|
-
/**
|
|
645
|
-
* @generated from enum value: BF16 = 2;
|
|
646
|
-
*/
|
|
647
|
-
BF16 = 2,
|
|
648
|
-
/**
|
|
649
|
-
* @generated from enum value: F16 = 3;
|
|
650
|
-
*/
|
|
651
|
-
F16 = 3
|
|
652
|
-
}
|
|
653
|
-
/**
|
|
654
|
-
* Describes the enum jammi.v1.fine_tune.BackboneDtype.
|
|
655
|
-
*/
|
|
656
|
-
export declare const BackboneDtypeSchema: GenEnum<BackboneDtype>;
|
|
657
|
-
/**
|
|
658
|
-
* FineTuneService — the wire surface for the engine's fine-tuning verb.
|
|
659
|
-
*
|
|
660
|
-
* `StartFineTune` and `FineTuneStatus` are imperative session methods, not SQL
|
|
661
|
-
* queries, so they live as typed gRPC RPCs alongside SessionService /
|
|
662
|
-
* TriggerService / EmbeddingService / InferenceService / EvalService rather
|
|
663
|
-
* than on the Flight SQL surface (which carries query/result only, per ADR-01
|
|
664
|
-
* §3.2). Fine-tuning is a substrate primitive — training a LoRA adapter over a
|
|
665
|
-
* registered source — and names no tenant-domain concept.
|
|
666
|
-
*
|
|
667
|
-
* The two verbs delegate 1:1 to the consumer-facing `Session`/`LocalSession`
|
|
668
|
-
* abstraction: `StartFineTune` mirrors `Session::fine_tune` (which returns a
|
|
669
|
-
* `FineTuneJobId`) and `FineTuneStatus` mirrors `Session::fine_tune_status`.
|
|
670
|
-
* The job runs asynchronously inside the engine; the client polls
|
|
671
|
-
* `FineTuneStatus` by id until a terminal state. There is no progress stream —
|
|
672
|
-
* the abstraction exposes none, so this surface invents none. The service
|
|
673
|
-
* reimplements no training, LoRA, or scan logic; it is purely a wire adapter.
|
|
674
|
-
*
|
|
675
|
-
* Tenant scope is read from the request's `SessionTenant` extension, set by the
|
|
676
|
-
* shared TenantInterceptor from the `jammi-session-id` header — the same
|
|
677
|
-
* mechanism every other gRPC service and Flight SQL already use.
|
|
678
|
-
*
|
|
679
|
-
* @generated from service jammi.v1.fine_tune.FineTuneService
|
|
680
|
-
*/
|
|
681
|
-
export declare const FineTuneService: GenService<{
|
|
682
|
-
/**
|
|
683
|
-
* Start a fine-tuning job over a registered source and return its id. The
|
|
684
|
-
* job trains asynchronously; poll `FineTuneStatus` with the returned id.
|
|
685
|
-
* Peer of the abstraction's `fine_tune`.
|
|
686
|
-
*
|
|
687
|
-
* @generated from rpc jammi.v1.fine_tune.FineTuneService.StartFineTune
|
|
688
|
-
*/
|
|
689
|
-
startFineTune: {
|
|
690
|
-
methodKind: "unary";
|
|
691
|
-
input: typeof StartFineTuneRequestSchema;
|
|
692
|
-
output: typeof StartFineTuneResponseSchema;
|
|
693
|
-
};
|
|
694
|
-
/**
|
|
695
|
-
* Read a fine-tune job's current status by id. Peer of the abstraction's
|
|
696
|
-
* `fine_tune_status`.
|
|
697
|
-
*
|
|
698
|
-
* @generated from rpc jammi.v1.fine_tune.FineTuneService.FineTuneStatus
|
|
699
|
-
*/
|
|
700
|
-
fineTuneStatus: {
|
|
701
|
-
methodKind: "unary";
|
|
702
|
-
input: typeof FineTuneStatusRequestSchema;
|
|
703
|
-
output: typeof FineTuneStatusResponseSchema;
|
|
704
|
-
};
|
|
705
|
-
}>;
|
|
706
|
-
//# sourceMappingURL=fine_tune_pb.d.ts.map
|