ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add force sample toggle to the ui

Browse Source
This commit is contained in:
Jaret Burkett
2025-08-31 16:58:27 -06:00
parent 193c1b2dfa
commit 0f2239ca23
4 changed files with 42 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
ui/src/app/jobs/new/SimpleJob.tsx
View File

@@ -809,7 +809,28 @@ export default function SimpleJob({
label="Skip First Sample"
className="pt-4"
checked={jobConfig.config.process[0].train.skip_first_sample || false}
onChange={value => setJobConfig(value, 'config.process[0].train.skip_first_sample')}
onChange={value => {
setJobConfig(value, 'config.process[0].train.skip_first_sample');
// cannot do both, so disable the other
if (value){
setJobConfig(false, 'config.process[0].train.force_first_sample');
}
}}
/>
</div>
<div>
<Checkbox
label="Force First Sample"
className="pt-1"
checked={jobConfig.config.process[0].train.force_first_sample || false}
docKey={'train.force_first_sample'}
onChange={value => {
setJobConfig(value, 'config.process[0].train.force_first_sample');
// cannot do both, so disable the other
if (value){
setJobConfig(false, 'config.process[0].train.skip_first_sample');
}
}}
/>
</div>
<div>
@@ -817,7 +838,13 @@ export default function SimpleJob({
label="Disable Sampling"
className="pt-1"
checked={jobConfig.config.process[0].train.disable_sampling || false}
onChange={value => setJobConfig(value, 'config.process[0].train.disable_sampling')}
onChange={value => {
setJobConfig(value, 'config.process[0].train.disable_sampling');
// cannot do both, so disable the other
if (value){
setJobConfig(false, 'config.process[0].train.force_first_sample');
}
}}
/>
</div>
</FormGroup>

1
ui/src/app/jobs/new/jobConfig.ts
View File

@@ -75,6 +75,7 @@ export const defaultJobConfig: JobConfig = {
ema_decay: 0.99,
},
skip_first_sample: false,
force_first_sample: false,
disable_sampling: false,
dtype: 'bf16',
diff_output_preservation: false,

11
ui/src/docs.tsx
View File

@@ -155,6 +155,17 @@ const docs: { [key: string]: ConfigDoc } = {
</>
),
},
'train.force_first_sample': {
title: 'Force First Sample',
description: (
<>
This option will force the trainer to generate samples when it starts. The trainer will normally only generate a first sample
when nothing has been trained yet, but will not do a first sample when resuming from an existing checkpoint. This option
forces a first sample every time the trainer is started. This can be useful if you have changed sample prompts and want to see
the new prompts right away.
</>
),
},
};
export const getDoc = (key: string | null | undefined): ConfigDoc | null => {

1
ui/src/types.ts
View File

@@ -117,6 +117,7 @@ export interface TrainConfig {
weight_decay: number;
};
skip_first_sample: boolean;
force_first_sample: boolean;
disable_sampling: boolean;
diff_output_preservation: boolean;
diff_output_preservation_multiplier: number;