From c2d4eb25da282082f43e3cdd7f780c73bef0d5f4 Mon Sep 17 00:00:00 2001
From: ali asaria <aliasaria@users.noreply.github.com>
Date: Fri, 28 Jun 2024 13:57:41 -0400
Subject: [PATCH] remove default config from plugin

---
 .../Experiment/Train/TrainingModalLoRA.tsx         | 14 ++------------
 1 file changed, 2 insertions(+), 12 deletions(-)

diff --git a/src/renderer/components/Experiment/Train/TrainingModalLoRA.tsx b/src/renderer/components/Experiment/Train/TrainingModalLoRA.tsx
index 52c87074..b3998e4d 100644
--- a/src/renderer/components/Experiment/Train/TrainingModalLoRA.tsx
+++ b/src/renderer/components/Experiment/Train/TrainingModalLoRA.tsx
@@ -22,16 +22,6 @@ import {
 import DynamicPluginForm from '../DynamicPluginForm';
 import TrainingModalDataTab from './TraningModalDataTab';
 
-const DefaultLoraConfig = {
-  model_max_length: 2048,
-  num_train_epochs: 3,
-  learning_rate: 1e-3,
-  lora_r: 8,
-  lora_alpha: 16,
-  lora_dropout: 0.05,
-  adaptor_name: '',
-};
-
 import { generateFriendlyName } from 'renderer/lib/utils';
 const fetcher = (url) => fetch(url).then((res) => res.json());
 
@@ -71,7 +61,7 @@ export default function TrainingModalLoRA({
 }) {
   // Store the current selected Dataset in this modal
   const [selectedDataset, setSelectedDataset] = useState(null);
-  const [config, setConfig] = useState(DefaultLoraConfig);
+  const [config, setConfig] = useState({});
   const [nameInput, setNameInput] = useState('');
 
   // Fetch available datasets from the API
@@ -134,7 +124,7 @@ export default function TrainingModalLoRA({
     } else {
       //This case is for when we are creating a new template
       setSelectedDataset(null);
-      setConfig(DefaultLoraConfig);
+      setConfig({});
       setNameInput(generateFriendlyName());
     }
   }, [templateData]);
-- 
GitLab