网站首页
编程语言
数据库
系统相关
其他分享
编程问答
whystea5
2024-07-29
whystea5
加载模型及量化frommodelscopeimportAutoTokenizer,AutoModeltokenizer=AutoTokenizer.from_pretrained(model_dir,trust_remote_code=True)model=AutoModelForCausalLM.from_pretrained(model_dir,trust_remote_code=True).quantize(4).half().cuda()构造数据集