Save model path in training template for local models.

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;

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
                   />