diff --git a/src/renderer/components/Experiment/Train/LoRATrainingRunButton.tsx b/src/renderer/components/Experiment/Train/LoRATrainingRunButton.tsx index 9408eb17876fdaddefde493f1c028ecf9140cc4c..7cb531cf8fefd4a5d4d756b1b471d0fc32a2b63d 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 086d3376e7e05a3a1842eb99d5692f27eb8a8c20..214a3dd52ddb007708e92ae243c651ddecb6759b 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} />