diff --git a/jobs/process/ExtractLoconProcess.py b/jobs/process/ExtractLoconProcess.py index 08e9f00c..e0e16506 100644 --- a/jobs/process/ExtractLoconProcess.py +++ b/jobs/process/ExtractLoconProcess.py @@ -35,7 +35,7 @@ class ExtractLoconProcess(BaseExtractProcess): self.disable_cp = self.get_conf('disable_cp', False) # set modes - if self.mode not in ['fixed', 'threshold', 'ratio', 'quantile']: + if self.mode not in list(mode_dict.keys()): raise ValueError(f"Unknown mode: {self.mode}") self.linear_param = self.get_conf('linear', mode_dict[self.mode]['linear'], as_type=mode_dict[self.mode]['type']) self.conv_param = self.get_conf('conv', mode_dict[self.mode]['conv'], as_type=mode_dict[self.mode]['type']) diff --git a/jobs/process/ExtractLoraProcess.py b/jobs/process/ExtractLoraProcess.py index af1cdde4..70ea7e16 100644 --- a/jobs/process/ExtractLoraProcess.py +++ b/jobs/process/ExtractLoraProcess.py @@ -2,10 +2,23 @@ from collections import OrderedDict from toolkit.lycoris_utils import extract_diff from .BaseExtractProcess import BaseExtractProcess + mode_dict = { 'fixed': { - 'dim': 4, + 'dim': 64, 'type': int + }, + 'threshold': { + 'dim': 0, + 'type': float + }, + 'ratio': { + 'dim': 0.5, + 'type': float + }, + 'quantile': { + 'dim': 0.5, + 'type': float } } @@ -20,7 +33,7 @@ class ExtractLoraProcess(BaseExtractProcess): self.mode = self.get_conf('mode', 'fixed') # set modes - if self.mode not in ['fixed']: + if self.mode not in list(mode_dict.keys()): raise ValueError(f"Unknown mode: {self.mode}") self.dim = self.get_conf('dim', mode_dict[self.mode]['dim'], as_type=mode_dict[self.mode]['type']) self.use_sparse_bias = self.get_conf('use_sparse_bias', False)