From f70f9b296cdb71fc580af4dbe6ad2f933502c5ed Mon Sep 17 00:00:00 2001
From: Tony Salomone <dadmobile@gmail.com>
Date: Thu, 29 Feb 2024 11:57:11 -0500
Subject: [PATCH] Save model path in training template for local models.
---
src/renderer/components/Experiment/Train/TrainLoRA.tsx | 4 ++--
.../components/Experiment/Train/TrainingModalLoRA.tsx | 8 +++++---
2 files changed, 7 insertions(+), 5 deletions(-)
diff --git a/src/renderer/components/Experiment/Train/TrainLoRA.tsx b/src/renderer/components/Experiment/Train/TrainLoRA.tsx
index e25415d1..45e8a472 100644
--- a/src/renderer/components/Experiment/Train/TrainLoRA.tsx
+++ b/src/renderer/components/Experiment/Train/TrainLoRA.tsx
@@ -64,9 +64,9 @@ function formatJobConfig(c): ReactElement {
{/* {JSON.stringify(c)} */}
<b>Template ID:</b> {c.template_name}
<br />
- <b>Model Name:</b> {c.model_name}
+ <b>Model:</b> {c.model_name}
<br />
- <b>Dataset Name:</b> {c.dataset_name}
+ <b>Dataset:</b> {c.dataset_name}
</>
);
return r;
diff --git a/src/renderer/components/Experiment/Train/TrainingModalLoRA.tsx b/src/renderer/components/Experiment/Train/TrainingModalLoRA.tsx
index 153809b5..3b46a5af 100644
--- a/src/renderer/components/Experiment/Train/TrainingModalLoRA.tsx
+++ b/src/renderer/components/Experiment/Train/TrainingModalLoRA.tsx
@@ -78,7 +78,9 @@ export default function TrainingModalLoRA({ open, onClose, experimentInfo }) {
return chatAPI.Endpoints.Dataset.Info(selectedDataset);
}, fetcher);
- const currentModelName = experimentInfo?.config?.foundation;
+ const currentModel = experimentInfo?.config?.foundation_filename ?
+ experimentInfo?.config?.foundation_filename :
+ experimentInfo?.config?.foundation;
function injectIntoTemplate(key) {
// Add the key to the textbox with id "template"
@@ -182,7 +184,7 @@ export default function TrainingModalLoRA({ open, onClose, experimentInfo }) {
<Stack direction="row" justifyContent="space-evenly" gap={2}>
<FormControl sx={{ flex: 1 }}>
<FormLabel>Model:</FormLabel>
- <Typography variant="soft">{currentModelName}</Typography>
+ <Typography variant="soft">{currentModel}</Typography>
</FormControl>
<FormControl sx={{ flex: 1 }}>
<FormLabel>Architecture:</FormLabel>
@@ -193,7 +195,7 @@ export default function TrainingModalLoRA({ open, onClose, experimentInfo }) {
<input
hidden
- value={currentModelName}
+ value={currentModel}
name="model_name"
readOnly
/>
--
GitLab