|
@@ -2,8 +2,7 @@
|
|
|
# This software may be used and distributed according to the terms of the Llama 2 Community License Agreement.
|
|
|
|
|
|
import inspect
|
|
|
-from dataclasses import fields
|
|
|
-
|
|
|
+from dataclasses import asdict
|
|
|
from peft import (
|
|
|
LoraConfig,
|
|
|
AdaptionPromptConfig,
|
|
@@ -45,7 +44,7 @@ def generate_peft_config(train_config, kwargs):
|
|
|
config = configs[names.index(train_config.peft_method)]()
|
|
|
|
|
|
update_config(config, **kwargs)
|
|
|
- params = {k.name: getattr(config, k.name) for k in fields(config)}
|
|
|
+ params = asdict(config)
|
|
|
peft_config = peft_configs[names.index(train_config.peft_method)](**params)
|
|
|
|
|
|
return peft_config
|