From 41573f3a35b3ff6650c798fc22cc9617cb27be04 Mon Sep 17 00:00:00 2001
From: ali asaria <aliasaria@users.noreply.github.com>
Date: Thu, 20 Jun 2024 10:38:26 -0400
Subject: [PATCH] training ui updates
---
.../components/Experiment/Train/TrainLoRA.tsx | 21 +++++++++++++++++--
1 file changed, 19 insertions(+), 2 deletions(-)
diff --git a/src/renderer/components/Experiment/Train/TrainLoRA.tsx b/src/renderer/components/Experiment/Train/TrainLoRA.tsx
index 59a2caa2..58a55aa0 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>
))}
--
GitLab