From d17bde7fc8241c96b713d91bb57463919830540e Mon Sep 17 00:00:00 2001
From: deep1401 <gandhi0869@gmail.com>
Date: Wed, 5 Mar 2025 12:17:55 -0800
Subject: [PATCH] Remove model download prompts when training with no
foundation and improve listing of tasks table
---
.../components/Experiment/Train/LoRATrainingRunButton.tsx | 5 +++++
src/renderer/components/Experiment/Train/TrainLoRA.tsx | 2 +-
2 files changed, 6 insertions(+), 1 deletion(-)
diff --git a/src/renderer/components/Experiment/Train/LoRATrainingRunButton.tsx b/src/renderer/components/Experiment/Train/LoRATrainingRunButton.tsx
index 0c085aaf..67ebfe72 100644
--- a/src/renderer/components/Experiment/Train/LoRATrainingRunButton.tsx
+++ b/src/renderer/components/Experiment/Train/LoRATrainingRunButton.tsx
@@ -49,11 +49,16 @@ export default function LoRATrainingRunButton({
return false;
});
let modelInLocalList = false;
+ if (model === "unknown")
+ {
+ modelInLocalList = true;
+ } else {
models_downloaded.forEach(modelData => {
if (modelData.model_id == model || modelData.local_path === model) {
modelInLocalList = true;
}
});
+ }
const datasets_downloaded = await fetch(
chatAPI.Endpoints.Dataset.LocalList()
diff --git a/src/renderer/components/Experiment/Train/TrainLoRA.tsx b/src/renderer/components/Experiment/Train/TrainLoRA.tsx
index d09bd5db..8082caa6 100644
--- a/src/renderer/components/Experiment/Train/TrainLoRA.tsx
+++ b/src/renderer/components/Experiment/Train/TrainLoRA.tsx
@@ -63,7 +63,7 @@ function formatTemplateConfig(config): ReactElement {
const r = (
<>
- <b>Model:</b> {short_model_name} <br />
+ {short_model_name && (<><b>Model:</b> {short_model_name} <br /></>)}
<b>Dataset:</b> {c.dataset_name} <FileTextIcon size={14} />
<br />
{/* <b>Adaptor:</b> {c.adaptor_name} <br /> */}
--
GitLab