ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-11 08:20:35 +00:00
Code Issues 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}
onChange={value => setJobConfig(value, 'config.process[0].train.optimizer')}
options={[
{ value: 'adamw8bit', label: 'AdamW8Bit' },
{ 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