赞
踩
- def get_peft_model(model: PreTrainedModel, peft_config: PeftConfig, adapter_name: str = "default") -> PeftModel:
- """
- Returns a Peft model object from a model and a config.
- Args:
- model ([`transformers.PreTrainedModel`]): Model to be wrapped.
- peft_config ([`PeftConfig`]): Configuration object containing the parameters of the Peft model.
- """
- model_config = getattr(model, "config", {"model_type": "custom"})
- if hasattr(model_config, "to_dict"):
- model_config = model_config.to_dict()
-
- peft_config.base_model_name_or_path = model.__dict__.get("name_or_path", None)
-
- if peft_config.task_type not in MODEL_TYPE_TO_PEFT_MODEL_MAPPING.keys() and not peft_config.is_prompt_learning:
- return PeftModel(model, peft_config, adapte
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。