Skip to content

Commit 035c156

Browse files
committed
Avoid overwriting trainedWords in training
1 parent 1d623c8 commit 035c156

File tree

3 files changed

+2
-4
lines changed

3 files changed

+2
-4
lines changed

src/components/Resource/Forms/ModelVersionUpsertForm.tsx

+1-1
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ const schema = modelVersionUpsertSchema2
8585
.array()
8686
.optional(),
8787
})
88-
.refine((data) => (!data.skipTrainedWords ? data.trainedWords.length > 0 : true), {
88+
.refine((data) => (!data.skipTrainedWords ? (data.trainedWords ?? []).length > 0 : true), {
8989
message: 'You need to specify at least one trained word',
9090
path: ['trainedWords'],
9191
})

src/components/Resource/Forms/TrainingSelectFile.tsx

-2
Original file line numberDiff line numberDiff line change
@@ -201,8 +201,6 @@ export default function TrainingSelectFile({
201201
modelId: model.id,
202202
name: modelVersion.name,
203203
baseModel: modelVersion.baseModel,
204-
trainedWords: modelVersion.trainedWords,
205-
// ---
206204
trainingStatus: TrainingStatus.Approved,
207205
};
208206

src/server/schema/model-version.schema.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,7 @@ export const modelVersionUpsertSchema2 = z.object({
193193
epochs: z.number().min(0).max(100000).nullish(),
194194
clipSkip: z.number().min(1).max(12).nullish(),
195195
vaeId: z.number().nullish(),
196-
trainedWords: z.array(z.string()).default([]),
196+
trainedWords: z.array(z.string()).optional(),
197197
trainingStatus: z.nativeEnum(TrainingStatus).nullish(),
198198
trainingDetails: trainingDetailsObj.nullish(),
199199
status: z.nativeEnum(ModelStatus).optional(),

0 commit comments

Comments
 (0)