ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-03-10 21:19:49 +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,