diff --git a/src/renderer/components/Experiment/Train/TrainLoRA.tsx b/src/renderer/components/Experiment/Train/TrainLoRA.tsx
index e25415d14a143af755ac1271b09e141bf3b3088d..45e8a472a0169df1bd24ce8cfa36172e28d0b57d 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 153809b544439655edab06e52ad9ca04d1fdafcf..3b46a5afabcd15c2d9f32fccb94ab8ffeafec2b5 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
                   />