From 2514369a831223d26a9a11c5b85dcab8b9942b6b Mon Sep 17 00:00:00 2001
From: deep1401 <gandhi0869@gmail.com>
Date: Wed, 5 Mar 2025 11:45:32 -0800
Subject: [PATCH] ENable training when no plugin is loaded, enable running a
plugin if it has no parameter called model architectures
---
src/renderer/components/Experiment/Train/TrainLoRA.tsx | 6 ++++--
src/renderer/components/Nav/Sidebar.tsx | 4 +---
2 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/src/renderer/components/Experiment/Train/TrainLoRA.tsx b/src/renderer/components/Experiment/Train/TrainLoRA.tsx
index 83ff9d34..d09bd5db 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 d996748e..0b6671ab 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"
--
GitLab