Qwen2-VL 视觉大模型 快速 Qwen2-VL-7B-Instruct部署

Qwen2-VL 视觉大模型Qwen2-VL-7B-Instruct部署

注:本文完全参考b站博主:小林绿子的怀中猫 讲解内容,文章的图片几乎完全出自其讲解视频,我只是对其进行了注释,此篇博客完全是为了记录操作流程方便后续查看学习,如有侵权联系我删除。

b站视频:1.从零部署qwen2-vl-7b-Instruct模型_哔哩哔哩_bilibili

  • 环境(在AutoDL上进行快速部署)

计费方式选择按量计费,地区选择V100专区,GPU型号选择V100-32GB,GPU数量选择1台,如图中红框所示任意点击可租用的服务器。

选择基础镜像,pyTorch/2.3.0/3.12(ubuntu22.04)/12.1,然后点击立即创建。

当服务器创建好后,点击JupyterLab

要把下载的内容放到数据盘(autodl-tmp)中,否则万一下错会占用系统盘很多内存。

  • 模型下载

点击autodl-tmp文件夹,在其中创建download.py,并在其中编写模型下载代码。

上述红框中代码内容来自ModelScope,其模型下载地址为:ModelScope 魔搭社区,点击该下载地址,进入以下界面,该界面为Qwen2-VL-7B-Instruct相关内容下载界面,点击下载模型,赋值 SDK下载条例中内容复制到download.py文件中即可,然后CTRL+S保存即可。

在autodl-tmp文件夹下创建终端,确保在autodl-tmp文件夹下,如图中1所示,点击2出+号,在弹出界面中点击终端(如图中3处所示),终端创建成功。

点击终端后,弹出如下界面:

首先下载modelscope,复制:pip install modelscope至终端对话框,后按回车键即可运行。下载模型如果速度太慢可进行学术加速。点击帮助文档。

点击学术自选加速,复制红框中的代码至终端界面执行。

然后再运行download.py

  • 项目代码与安装

https://github.com/QwenLM/Qwen2-VL

在autoal-tmp中创建一个文件夹,命名为Qwen;

然后跑进入Qwen的文件夹再打开一个终端。

然后将下列代码复制到新建终端运行;

git clone https://github.com/QwenLM/Qwen2-VL

cd Qwen2-VL

pip install qwen-vl-utils[decord]

pip install transformers

pip install 'accelerate>=0.26.0'

可能会报错,如下:

此处报错说明pip三行代码下载成功,项目Qwen2-VL没有安装成功;

可在使用学术加速,即每新开一个终端都要使用学术加速。source /etc/network_turbo

然后在终端命令行输入:git clone https://github.com/QwenLM/Qwen2-VL进行下载

安装完成后在Qwen/Qwen2-VL文件夹中创建imgs,并点击上传图片按钮上传传本地下载图片。

判断模型是否下载成功查看,若出现Download model successfully则表示模型下载成功。否则表示没有下载成功。

  • 移动模型

一般建议放在数据盘(因为没有足够把握,若出现错误放到系统盘会爆)

此处要特别注意自己模型所下载的路径,以下为自己模型路径(若界面上没有显示下面图片红框中的路径,可以在执行一下python dowonload.py 会提示已经下载好模型,并显示出下载路径

然后移动路径:mv /root/.cache/modelscope/hub/Qwen/Qwen2-VL-7B-Instruct /root/autodl-tmp/Qwen/;

上述将模型移动至与Qwen2-VL同级目录

移动好后该文件下应该包含下述文件

然后在项目文件中新建一个文件test.py

并在test.py中粘贴下列内容,有三处需要修改,分别为模型路径和图片存储路径,用红色框出,其中模型路径一定和上述保持一致(移动后的)。

from transformers import Qwen2VLForConditionalGeneration, AutoTokenizer, AutoProcessor

from qwen_vl_utils import process_vision_info

# default: Load the model on the available device(s)

model = Qwen2VLForConditionalGeneration.from_pretrained(

    "/root/autodl-tmp/Qwen/Qwen2-VL-7B-Instruct", torch_dtype="auto", device_map="auto"

)

# We recommend enabling flash_attention_2 for better acceleration and memory saving, especially in multi-image and video scenarios.

# model = Qwen2VLForConditionalGeneration.from_pretrained(

#     "Qwen/Qwen2-VL-7B-Instruct",

#     torch_dtype=torch.bfloat16,

#     attn_implementation="flash_attention_2",

#     device_map="auto",

# )

# default processer

processor = AutoProcessor.from_pretrained("/root/autodl-tmp/Qwen/Qwen2-VL-7B-Instruct ")

# The default range for the number of visual tokens per image in the model is 4-16384.

# You can set min_pixels and max_pixels according to your needs, such as a token range of 256-1280, to balance performance and cost.

# min_pixels = 256*28*28

# max_pixels = 1280*28*28

# processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct", min_pixels=min_pixels, max_pixels=max_pixels)

messages = [

    {

        "role": "user",

        "content": [

            {

                "type": "image",

                "image": "imgs/1.png",

            },

            {"type": "text", "text": "Describe this image."},

        ],

    }

]

# Preparation for inference

text = processor.apply_chat_template(

    messages, tokenize=False, add_generation_prompt=True

)

image_inputs, video_inputs = process_vision_info(messages)

inputs = processor(

    text=[text],

    images=image_inputs,

    videos=video_inputs,

    padding=True,

    return_tensors="pt",

)

inputs = inputs.to("cuda")

# Inference: Generation of the output

generated_ids = model.generate(**inputs, max_new_tokens=128)

generated_ids_trimmed = [

    out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)

]

output_text = processor.batch_decode(

    generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False

)

print(output_text)

对图片进行提问修改test.py中如下内容,可根据自己的需求提问

点击ctrl+s保存,然后进入终端3(项目文件所创建的终端)输入python test.py运行

运行结果如下:

7B模型的推理能力还有待加强;

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值