diff --git a/src/renderer/components/Experiment/Train/TrainLoRA.tsx b/src/renderer/components/Experiment/Train/TrainLoRA.tsx index 59a2caa20d93c43a4a604b6ba125a32c36edfab3..58a55aa071ff005d3a1a2bbc7ed16d7944ff663b 100644 --- a/src/renderer/components/Experiment/Train/TrainLoRA.tsx +++ b/src/renderer/components/Experiment/Train/TrainLoRA.tsx @@ -12,6 +12,7 @@ import { Dropdown, IconButton, LinearProgress, + ListItemDecorator, Menu, MenuButton, MenuItem, @@ -25,7 +26,9 @@ import { FileTextIcon, GraduationCapIcon, LineChartIcon, + Plug2Icon, PlusCircleIcon, + PlusIcon, Trash2Icon, } from 'lucide-react'; @@ -153,10 +156,20 @@ export default function TrainLoRA({ experimentInfo }) { </Typography> <Dropdown> - <MenuButton startDecorator={<PlusCircleIcon />} variant="solid"> + <MenuButton + color="primary" + size="sm" + startDecorator={<PlusIcon />} + variant="solid" + > New </MenuButton> - <Menu> + <Menu sx={{ maxWidth: '300px' }}> + <MenuItem disabled variant="soft" color="primary"> + <Typography level="title-sm"> + Select a training plugin from the following list: + </Typography> + </MenuItem> {pluginsData?.map((plugin) => ( <MenuItem onClick={() => { @@ -164,7 +177,11 @@ export default function TrainLoRA({ experimentInfo }) { setCurrentPlugin(plugin.uniqueId); setOpen(true); }} + key={plugin.uniqueId} > + <ListItemDecorator> + <Plug2Icon /> + </ListItemDecorator> {plugin.name} </MenuItem> ))}