ModelScope如何跑大模型

安装Aanconda

一般两个选择,一个是官网,另一个是国内镜像网站(大家自行选择)。前者大家都懂,速度感人,所以国内一般选择后者镜像下载。
1)官网,进去都是英文,不懂直接点翻译。
https://www.anaconda.com/download
在这里插入图片描述
看红色框子里面,选择这个就🆗。
在这里插入图片描述
2)我比较推荐大家用这种方法,自己使用的也是这种方法。(清华大学开源镜像网站)

https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

根据自己电脑选择对应版本:下载还是挺快的。
在这里插入图片描述
在这里插入图片描述

Anaconda的安装:

在这里插入图片描述
这里跳了一个“I agree”我想大家都知道,然后是注意这里:它默认是第一个!!!你选择All Users,然后next。
在这里插入图片描述
在这里插入图片描述
**!!!这里也是要注意的一个点:**第一个选项意思就是自动配置环境变量(觉得手动配置有难度的当然也可以直接勾选第一个,自动配置)

第二个选项是自己手动配置环境变量(后面有教程)。

第二个勾选默认的不用管。直接点击 Install
在这里插入图片描述

安装通义千问-1.8B

本想按照教程进行安装,执行到命令

pip install transformers==4.32.0 accelerate tiktoken einops scipy transformers_stream_generator==0.0.4 peft deepspeed

控制台提示需要cuda环境
在这里插入图片描述
于是尝试以下方案。
为了避免其他的环境受影响,使用conda创新一个新环境。

创建环境

conda create -n chat python==3.9

激活环境

conda activate chat

配置环境

pip3 install torch torchvision torchaudio
pip3 install modelscope
pip3 install -U transformers
pip3 install accelerate 

跑代码

from modelscope import AutoModelForCausalLM, AutoTokenizer
# 根据自己的设备调整device为cpu 或 cuda
device = "cpu" # the device to load the model onto

model = AutoModelForCausalLM.from_pretrained(
    "qwen/Qwen1.5-0.5B-Chat",
    device_map="auto"
)
tokenizer = AutoTokenizer.from_pretrained("qwen/Qwen1.5-0.5B-Chat")

prompt = "Give me a short introduction to large language model."
messages = [
    {"role": "system", "content": "You are a helpful assistant."},
    {"role": "user", "content": prompt}
]
text = tokenizer.apply_chat_template(
    messages,
    tokenize=False,
    add_generation_prompt=True
)
model_inputs = tokenizer([text], return_tensors="pt").to(device)

generated_ids = model.generate(
    model_inputs.input_ids,
    max_new_tokens=512
)
generated_ids = [
    output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
]
 
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
print(response)

运行结果

在这里插入图片描述

  • 7
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值