部署DeepSeek-R1-Distill-Qwen-7B并进行推理

devtools/2025/2/8 5:58:17/

下载模型文件:

将https://huggingface.co/deepseek-ai/DeepSeek-R1-Distill-Qwen-7B/tree/main目录下面的所有文件全部下载下来,稍微有点大,所有文件将近16个G

编写代码进行推理

$ more testDS.py 
from transformers import AutoModelForCausalLM, AutoTokenizer
from datetime import datetime
# 加载预训练的模型和分词器
model_name = '/data/model/DeepSeek-R1-Distill-Qwen-7B'
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)# 编写提示词
prompt = """"""# 对输入进行编码
input_ids = tokenizer.encode(prompt + tokenizer.eos_token, return_tensors='pt')
print("当前日期和时间:", datetime.now())
# 生成回复
output = model.generate(input_ids, max_length=10000, pad_token_id=tokenizer.eos_token_id, no_repeat_ngram_size=2, top_k=50, top_p=0.95, temperature=0.7)# 解码生成的文本
response = tokenizer.decode(output[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
print(response)
print("当前日期和时间:", datetime.now())

推理结果

$ python testDS.py 
Loading checkpoint shards: 100%|████████████████████████████████████████████████████████████████████████████████████████████| 2/2 [00:02<00:00,  1.23s/it]
当前日期和时间: 2025-02-07 14:40:02.670872
The attention mask is not set and cannot be inferred from input because pad token is same as eos token.As a consequence, you may observe unexpected behavior. Please pass your input's `attention_mask` to obtain reliable results.
<think>
xxx
</think>
领导人
当前日期和时间: 2025-02-07 14:40:57.471051


http://www.ppmy.cn/devtools/157026.html

相关文章

基于keepalived+GTID半同步主从复制的高可用MySQL集群

文章目录 项目架构图项目名称项目环境项目描述ip地址规划项目步骤一.安装好8台全新的centos7.9的系统&#xff0c;关闭firewalld和selinux&#xff0c;配置每台主机的静态ip地址&#xff0c;设置每台主机对应的主机名。1、关闭firewalld2.关闭seLinux3.配置每台主机静态ip地址4…

测试驱动开发(TDD)实践:从理论到实践

测试驱动开发&#xff08;Test-Driven Development&#xff0c;简称TDD&#xff09;是一种软件开发方法&#xff0c;其核心理念是“先写测试&#xff0c;再写代码”。与传统的开发方式不同&#xff0c;TDD并非开发完成后再进行测试&#xff0c;而是将测试置于开发的前沿&#x…

QT通过setProperty设置不同QSS样式

如上切换效果就是通过setProperty来实现切换不同颜色的。 实现以上效果第一步&#xff0c;需要在QSS中做属性处理。 QLabel{color:red;} QLabel[status"1"]{color:black;} QLabel[status"2"]{color:white;} QLabel[status"3"]{color:blue;} QLa…

centos系统清理docker日志文件

在CentOS系统中&#xff0c;Docker日志文件的不断增长可能会占用大量磁盘空间&#xff0c;影响系统性能和正常运行。为了保持系统的健康状态&#xff0c;需要定期清理Docker日志文件。本文将详细介绍如何在CentOS系统上清理Docker日志文件的具体步骤和方法。 Docker日志文件位…

GitHub 使用教程:从入门到进阶

1. GitHub账号注册 访问 GitHub 官网 (https://github.com)点击 “Sign up” 按钮填写用户名、邮箱和密码验证邮箱完成注册 2. 基础配置 2.1 安装Git 访问 Git 官网下载安装包运行安装程序&#xff0c;按提示完成安装打开终端&#xff0c;设置用户信息&#xff1a; git co…

基于知乎平台的“开源AI智能名片2 + 1链动模式S2B2C商城小程序”引流策略研究

摘要&#xff1a;本文聚焦于如何借助知乎平台的高权重及优质用户特性&#xff0c;对“开源AI智能名片2 1链动模式S2B2C商城小程序”进行有效引流。通过深入分析知乎平台的用户特点、引流规则&#xff0c;并结合具体的引流方法&#xff0c;旨在为相关项目在知乎平台实现高效用户…

云计算中的DevOps是什么?为什么它很重要?

是否曾在开发新产品时觉得时间过得飞快&#xff0c;往往在新功能准备好发布之前&#xff0c;系统却已经陷入了运行不稳定的怪圈&#xff1f;或者&#xff0c;感觉开发和运维团队之间总像两座遥远的孤岛&#xff0c;沟通困难&#xff0c;协作不畅&#xff0c;发布一次新版本&…

docker 安装 mindoc

文章目录 一、官网地址二、安装 一、官网地址 https://mindoc.com.cn/docs/mindochelp/mindoc-summary二、安装 docker run -it --namemindoc --restartalways -v /opt/mindoc-docker/conf:/mindoc/conf -v /opt/mindoc-docker/uploads:/mindoc/uploads -v/opt/mindoc-docker…