From 0be86d452ec13af987d500314425d8ad0fd9bb1a Mon Sep 17 00:00:00 2001
From: Tony Salomone <dadmobile@gmail.com>
Date: Thu, 11 Jul 2024 13:55:27 -0400
Subject: [PATCH] Add full training template config to job.job_data when
starting a train
---
.../Experiment/Train/LoRATrainingRunButton.tsx | 6 ++----
.../components/Experiment/Train/TrainLoRA.tsx | 12 ++++++++++--
2 files changed, 12 insertions(+), 6 deletions(-)
diff --git a/src/renderer/components/Experiment/Train/LoRATrainingRunButton.tsx b/src/renderer/components/Experiment/Train/LoRATrainingRunButton.tsx
index 9408eb17..7cb531cf 100644
--- a/src/renderer/components/Experiment/Train/LoRATrainingRunButton.tsx
+++ b/src/renderer/components/Experiment/Train/LoRATrainingRunButton.tsx
@@ -9,14 +9,12 @@ import { PlayIcon } from 'lucide-react';
export default function LoRATrainingRunButton({
initialMessage,
action = () => {},
- trainingTemplateId,
+ trainingTemplate,
jobsMutate,
experimentId,
}) {
const [progress, setProgress] = useState(0);
- let job_data = {
- template_id: trainingTemplateId,
- };
+ let job_data = trainingTemplate;
return (
<Button
variant="solid"
diff --git a/src/renderer/components/Experiment/Train/TrainLoRA.tsx b/src/renderer/components/Experiment/Train/TrainLoRA.tsx
index 086d3376..214a3dd5 100644
--- a/src/renderer/components/Experiment/Train/TrainLoRA.tsx
+++ b/src/renderer/components/Experiment/Train/TrainLoRA.tsx
@@ -219,7 +219,9 @@ export default function TrainLoRA({ experimentInfo }) {
<td>error...</td>
</tr>
)}
- {data &&
+ { // Format of template data by column:
+ // 0 = id, 1 = name, 2 = description, 3 = type, 4 = datasets, 5 = config, 6 = created, 7 = updated
+ data &&
data?.map((row) => {
return (
<tr key={row[0]}>
@@ -237,7 +239,13 @@ export default function TrainLoRA({ experimentInfo }) {
<ButtonGroup sx={{ justifyContent: 'flex-end' }}>
<LoRATrainingRunButton
initialMessage="Queue"
- trainingTemplateId={row[0]}
+ trainingTemplate={{
+ template_id: row[0],
+ template_name: row[1],
+ model_name: row[5]?.model_name || "unknown",
+ dataset: row[4],
+ config: row[5]
+ }}
jobsMutate={jobsMutate}
experimentId={experimentInfo?.id}
/>
--
GitLab