diff --git a/src/renderer/components/Experiment/Train/TrainLoRA.tsx b/src/renderer/components/Experiment/Train/TrainLoRA.tsx index 83ff9d34a4ed24c2e03e22fe1773632cc8e9952e..d09bd5db07a2a4e4010a8e8df997298f111204a1 100644 --- a/src/renderer/components/Experiment/Train/TrainLoRA.tsx +++ b/src/renderer/components/Experiment/Train/TrainLoRA.tsx @@ -232,7 +232,9 @@ export default function TrainLoRA({ experimentInfo }) { }} key={plugin.uniqueId} disabled={ - !plugin.model_architectures?.includes(modelArchitecture) + plugin.model_architectures + ? !plugin.model_architectures.includes(modelArchitecture) + : false } > <ListItemDecorator> @@ -244,7 +246,7 @@ export default function TrainLoRA({ experimentInfo }) { level="body-xs" sx={{ color: 'var(--joy-palette-neutral-400)' }} > - {!plugin.model_architectures?.includes(modelArchitecture) + {plugin.model_architectures && !plugin.model_architectures.includes(modelArchitecture) ? '(Does not support this model architecture)' : ''} </Typography> diff --git a/src/renderer/components/Nav/Sidebar.tsx b/src/renderer/components/Nav/Sidebar.tsx index d996748ea952b341ac959212d468ebcf2959d155..0b6671ab81fdac71e784cc41ef20e4579aeb2e0f 100644 --- a/src/renderer/components/Nav/Sidebar.tsx +++ b/src/renderer/components/Nav/Sidebar.tsx @@ -164,9 +164,7 @@ export default function Sidebar({ title="Train" path="/projects/training" icon={<GraduationCapIcon />} - disabled={ - !experimentInfo?.name || !experimentInfo?.config?.foundation - } + disabled={!experimentInfo?.name} /> <SubNavItem title="Export"