Skip to content
Snippets Groups Projects
Commit c2d4eb25 authored by ali asaria's avatar ali asaria
Browse files

remove default config from plugin

parent fb4d2f23
Branches
Tags
No related merge requests found
...@@ -22,16 +22,6 @@ import { ...@@ -22,16 +22,6 @@ import {
import DynamicPluginForm from '../DynamicPluginForm'; import DynamicPluginForm from '../DynamicPluginForm';
import TrainingModalDataTab from './TraningModalDataTab'; 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'; import { generateFriendlyName } from 'renderer/lib/utils';
const fetcher = (url) => fetch(url).then((res) => res.json()); const fetcher = (url) => fetch(url).then((res) => res.json());
...@@ -71,7 +61,7 @@ export default function TrainingModalLoRA({ ...@@ -71,7 +61,7 @@ export default function TrainingModalLoRA({
}) { }) {
// Store the current selected Dataset in this modal // Store the current selected Dataset in this modal
const [selectedDataset, setSelectedDataset] = useState(null); const [selectedDataset, setSelectedDataset] = useState(null);
const [config, setConfig] = useState(DefaultLoraConfig); const [config, setConfig] = useState({});
const [nameInput, setNameInput] = useState(''); const [nameInput, setNameInput] = useState('');
// Fetch available datasets from the API // Fetch available datasets from the API
...@@ -134,7 +124,7 @@ export default function TrainingModalLoRA({ ...@@ -134,7 +124,7 @@ export default function TrainingModalLoRA({
} else { } else {
//This case is for when we are creating a new template //This case is for when we are creating a new template
setSelectedDataset(null); setSelectedDataset(null);
setConfig(DefaultLoraConfig); setConfig({});
setNameInput(generateFriendlyName()); setNameInput(generateFriendlyName());
} }
}, [templateData]); }, [templateData]);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment