diff --git a/src/renderer/components/Experiment/Train/LoRATrainingRunButton.tsx b/src/renderer/components/Experiment/Train/LoRATrainingRunButton.tsx index 0c085aaf62416fb32d0906c9dd3823985f54905b..67ebfe720470abd7ca336d5aad0c8d6b4bcf89df 100644 --- a/src/renderer/components/Experiment/Train/LoRATrainingRunButton.tsx +++ b/src/renderer/components/Experiment/Train/LoRATrainingRunButton.tsx @@ -49,11 +49,16 @@ export default function LoRATrainingRunButton({ return false; }); let modelInLocalList = false; + if (model === "unknown") + { + modelInLocalList = true; + } else { models_downloaded.forEach(modelData => { if (modelData.model_id == model || modelData.local_path === model) { modelInLocalList = true; } }); + } const datasets_downloaded = await fetch( chatAPI.Endpoints.Dataset.LocalList() diff --git a/src/renderer/components/Experiment/Train/TrainLoRA.tsx b/src/renderer/components/Experiment/Train/TrainLoRA.tsx index d09bd5db07a2a4e4010a8e8df997298f111204a1..8082caa661b34c68c7d867b1eaf39f10582a3d2d 100644 --- a/src/renderer/components/Experiment/Train/TrainLoRA.tsx +++ b/src/renderer/components/Experiment/Train/TrainLoRA.tsx @@ -63,7 +63,7 @@ function formatTemplateConfig(config): ReactElement { const r = ( <> - <b>Model:</b> {short_model_name} <br /> + {short_model_name && (<><b>Model:</b> {short_model_name} <br /></>)} <b>Dataset:</b> {c.dataset_name} <FileTextIcon size={14} /> <br /> {/* <b>Adaptor:</b> {c.adaptor_name} <br /> */}