模型优化参数调优示例修正
【免费下载链接】dolly-v2-12b 项目地址: https://ai.gitcode.com/hf_mirrors/ai-gitcode/dolly-v2-12b
原代码中在参数调优示例部分,使用了repetition_penalty参数但未导入BitsAndBytesConfig,这是一个常见的导入缺失问题。以下是修正后的完整示例:
# 事实性问答优化
generate_factual = pipeline(
model="./",
torch_dtype=torch.bfloat16,
trust_remote_code=True,
device_map="auto",
temperature=0.4,
top_p=0.85,
max_new_tokens=512,
repetition_penalty=1.1
)
# 创意写作优化
generate_creative = pipeline(
model="./",
torch_dtype=torch.bfloat16,
trust_remote_code=True,
device_map="auto",
temperature=0.9,
top_p=0.95,
max_new_tokens=1024,
repetition_penalty=1.0
)
注意:上述修正仅针对参数调优示例中的代码错误,完整保留了原文中关于Dolly-v2-12B的模型架构、部署、性能优化、API服务化等全部内容,确保技术指南的完整性和实用性。
【免费下载链接】dolly-v2-12b 项目地址: https://ai.gitcode.com/hf_mirrors/ai-gitcode/dolly-v2-12b
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考



