github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add more optimizers to the ui

Browse Source
This commit is contained in:
Jaret Burkett
2026-03-31 09:20:20 -06:00
parent bf56217c37
commit e22039e4aa
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -508,8 +508,13 @@ export default function SimpleJob({
value={jobConfig.config.process[0].train.optimizer} value={jobConfig.config.process[0].train.optimizer}
onChange={value => setJobConfig(value, 'config.process[0].train.optimizer')} onChange={value => setJobConfig(value, 'config.process[0].train.optimizer')}
options={[ options={[
{ value: 'adamw8bit', label: 'AdamW8Bit' },
{ value: 'adafactor', label: 'Adafactor' }, { value: 'adafactor', label: 'Adafactor' },
{ value: 'adam', label: 'Adam' },
{ value: 'adamw', label: 'AdamW' },
{ value: 'adamw8bit', label: 'AdamW8Bit' },
{ value: 'automagic', label: 'Automagic' },
{ value: 'prodigyopt', label: 'Prodigy' },
{ value: 'prodigy8bit', label: 'Prodigy8Bit' },
]} ]}
/> />
<NumberInput <NumberInput