Skip to content

Commit c0e427c

Browse files
authored
Merge pull request #3627 from ZipRecruiter/mattGPT/fix-peft-config-load-bug-3623
fixes #3623: save PEFT config…
2 parents a7d1f8d + d8a7dd8 commit c0e427c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

flair/embeddings/transformer.py

+2
Original file line numberDiff line numberDiff line change
@@ -1148,6 +1148,7 @@ def is_supported_t5_model(config: PretrainedConfig) -> bool:
11481148
if "Please use the model as it is" not in str(e):
11491149
raise e
11501150

1151+
self.peft_config = peft_config
11511152
if peft_config is not None:
11521153
# add adapters for finetuning
11531154
try:
@@ -1396,6 +1397,7 @@ def to_params(self):
13961397
"subtoken_pooling": self.subtoken_pooling,
13971398
"cls_pooling": self.cls_pooling,
13981399
"config_state_dict": config_dict,
1400+
"peft_config": self.peft_config,
13991401
}
14001402

14011403
return model_state

0 commit comments

Comments
 (0)