对Peft的兼容性
为了支持习惯Peft的用户,Swift提供了对于Peft的兼容性。用户可以从swift中import peft组件:
PeftModel
PeftConfig
PeftModelForSeq2SeqLM
PeftModelForSequenceClassification
PeftModelForTokenClassification
PeftModelForCausalLM
PromptEncoderConfig
PromptTuningConfig
PrefixTuningConfig
PromptLearningConfig
LoraConfig
get_peft_config
get_peft_model_state_dict
get_peft_model
以上组件均可以从swift中import:
from swift import PeftModel, PeftConfig
Swift类也支持初始化Peft的tuner:
from modelscope.models.nlp import SbertForSequenceClassification
from modelscope.models.nlp.structbert import SbertConfig
from swift import LoraConfig, Swift
model = SbertForSequenceClassification(SbertConfig())
lora_config = LoraConfig(target_modules=['query', 'key', 'value'])
model = Swift.prepare_model(model, lora_config)
Swift对Peft进行了浅封装,使Peft可以在from_pretrained时使用modelscope hub中的模型。