Skip to content

Commit c1fe6c4

Browse files
HaohanTsaoyeonjoon-jung01
authored andcommitted
FIX & TEST: Fix GraLoRA bugs in get_peft_config_as_dict and improve test coverage
1 parent 9813b17 commit c1fe6c4

File tree

2 files changed

+551
-2
lines changed

2 files changed

+551
-2
lines changed

src/peft/tuners/gralora/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -228,8 +228,8 @@ def get_peft_config_as_dict(self, inference: bool = False):
228228
config = {k: v.value if isinstance(v, Enum) else v for k, v in asdict(value).items()}
229229
if inference:
230230
config["inference_mode"] = True
231-
config_dict[key] = config
232-
return config
231+
config_dict[key] = config
232+
return config_dict
233233

234234
def _set_adapter_layers(self, enabled=True):
235235
for module in self.model.modules():

0 commit comments

Comments
 (0)