模型优化参数调优示例修正

模型优化参数调优示例修正

【免费下载链接】dolly-v2-12b 【免费下载链接】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 【免费下载链接】dolly-v2-12b 项目地址: https://ai.gitcode.com/hf_mirrors/ai-gitcode/dolly-v2-12b

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值