We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c6b053d commit d8a7dd8Copy full SHA for d8a7dd8
flair/embeddings/transformer.py
@@ -1128,6 +1128,7 @@ def is_supported_t5_model(config: PretrainedConfig) -> bool:
1128
if "Please use the model as it is" not in str(e):
1129
raise e
1130
1131
+ self.peft_config = peft_config
1132
if peft_config is not None:
1133
# add adapters for finetuning
1134
try:
@@ -1376,6 +1377,7 @@ def to_params(self):
1376
1377
"subtoken_pooling": self.subtoken_pooling,
1378
"cls_pooling": self.cls_pooling,
1379
"config_state_dict": config_dict,
1380
+ "peft_config": self.peft_config,
1381
}
1382
1383
return model_state
0 commit comments