0


【“第二课堂”AI实践课堂笔记】:大模型学习

一、大模型介绍

大模型是人工智能领域中的一种重要模型,通常指的是具有大规模参数和复杂计算结构的机器学习模型。这些模型通常由深度神经网络构建而成,拥有数十亿甚至数千亿个参数,能够在多个领域和任务中展现出强大的泛化能力和自监督学习能力。以下是大模型的概念与特点:

1.1大模型的概念

大模型是指具有大规模参数和复杂计算结构的机器学习模型,它们通过训练海量数据来学习复杂的模式和特征,从而具备更强大的表示能力和泛化能力。大模型在自然语言处理、计算机视觉、语音识别等领域都有广泛的应用,是人工智能发展的重要方向之一。

1.2大模型的主要分类

  1. 自然语言处理(NLP)大模型:- 主要用于处理自然语言文本,如文本分类、命名实体识别、情感分析等。- 著名模型:GPT-3、BERT等。

  2. 计算机视觉(CV)大模型:- 主要用于处理图像和视频,如目标检测、语义分割、图像生成等。- 著名模型:Inception、ResNet、DenseNet等。

  3. 语音识别(ASR)大模型:- 主要用于语音信号的处理,如语音识别、语音合成等。- 著名模型:WaveNet、Transformer等。

  4. 推荐系统大模型:- 主要用于个性化推荐,如商品推荐、内容推荐等。- 著名算法:Collaborative Filtering、Content-Based Filtering等。

  5. 强化学习(RL)大模型:- 主要用于解决决策问题,如游戏、机器人等。- 著名模型:Q-learning、Deep Q-Network(DQN)等。

  6. 生成对抗网络(GAN)大模型:- 主要用于生成新的数据,如图像、音频、视频等。- 著名模型:DCGAN、StyleGAN等。

  7. 对话系统大模型:- 主要用于自然语言对话,如智能客服、智能助手等。- 著名模型:Rasa、Microsoft Turing-NLG等。

1.3大模型风险

1.数据泄露风险:大模型在训练过程中需要处理海量数据,这些数据往往包含用户的敏感信息。如果数据保护措施不到位,攻击者可能通过窃取数据或利用漏洞来访问敏感信息,导致数据泄露。

2.版权侵权:大模型在生成内容时,可能会侵犯他人的版权,如未经授权使用他人的作品作为训练数据或生成直接复制他人作品的内容。

3.模型误用:大模型在运营过程中可能会被误用,例如被用于生成恶意软件代码、进行网络攻击等。

二、使用驱动云构建千问模型

2.1环境方面准备

创建项目,选择B1.medium规格服务器

选择CUDA12.1 PyTorch2.1.1的镜像

在模型中搜索Qwen2-0.5B-Instruct

点击启动后,即可进入环境。

2.2文件上传

下载课前群中发布的大模型.zip文件

在/gemini/code中存放代码,/gemini/pretrain中是我们刚刚挂载的模型。

我们在本地解压大模型应用.zip,得到5个文件,进入/gemini/code目录,将文件拖入窗口完成上传。

2.3相关库的安装

  1. !pip install transformers==4.37.0 -U
  2. !pip install streamlit

使用pip安装来更新或安装特定版本的

  1. transformers

库用于处理和使用各种预训练的深度学习模型,如 BERT、GPT 和其他变换器

在Python环境中安装

  1. streamlit

库,

  1. streamlit

是一个非常流行的Python库,用于快速创建和共享数据应用。

2.4设置大模型流程

设置模型加载、分词器初始化、输入消息构建、模型输入准备以及生成文本的大部分流程。

  1. from transformers import AutoModelForCausalLM, AutoTokenizer
  2. device = "cuda" # the device to load the model onto
  3. model = AutoModelForCausalLM.from_pretrained(
  4. "/gemini/pretrain/Qwen2-0.5B-Instruct",
  5. torch_dtype="auto",
  6. device_map="auto"
  7. )
  8. tokenizer = AutoTokenizer.from_pretrained("/gemini/pretrain/Qwen2-0.5B-Instruct")
  9. prompt = "你是谁?"
  10. messages = [
  11. {"role": "system", "content": "You are a helpful assistant."},
  12. {"role": "user", "content": prompt}
  13. ]
  14. text = tokenizer.apply_chat_template(
  15. messages,
  16. tokenize=False,
  17. add_generation_prompt=True
  18. )
  19. model_inputs = tokenizer([text], return_tensors="pt").to(device)
  20. generated_ids = model.generate(
  21. model_inputs.input_ids,
  22. max_new_tokens=512
  23. )
  24. generated_ids = [
  25. output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
  26. ]
  27. response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
  28. print(respons

其中

  1. model = AutoModelForCausalLM.from_pretrained(
  2. "/gemini/pretrain/Qwen2-0.5B-Instruct",
  3. torch_dtype="auto",
  4. device_map="auto"
  5. )
  1. "/gemini/pretrain/Qwen2-0.5B-Instruct"

是模型的存储路径(可以是本地路径或模型名称)。

  1. torch_dtype="auto"

使得 PyTorch 自动选择适当的数据类型(例如,

  1. float32

  1. float16

)。

  1. device_map="auto"

使得模型自动映射到指定的设备(例如 GPU)

  1. generated_ids = model.generate(
  2. model_inputs.input_ids,
  3. max_new_tokens=512
  4. )

使用模型生成文本

  1. model_inputs.input_ids

是输入张量。

  1. max_new_tokens=512

限制生成的最大新 token 数量为 512。

2.5流式输出

接下来在test的基础上使用

  1. transformers

库中的模型进行流式文本生成,并处理多轮对话。通过使用线程,模型生成过程不会阻塞主线程,允许实时输出生成的文本。

  1. import torch
  2. from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
  3. from threading import Thread
  4. device = "cuda"
  5. model_name = "/gemini/pretrain/Qwen2-0.5B-Instruct"
  6. model = AutoModelForCausalLM.from_pretrained(model_name).to(device)
  7. tokenizer = AutoTokenizer.from_pretrained(model_name)
  8. streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
  9. messages = [
  10. {"role": "system", "content": "You are a helpful assistant."},
  11. ]
  12. /root/miniconda3/lib/python3.10/site-packages/tqdm/auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html
  13. from .autonotebook import tqdm as notebook_tqdm
  14. Special tokens have been added in the vocabulary, make sure the associated word embeddings are fine-tuned or trained.
  15. In [2]:
  16. def stream_generate(prompt, model, tokenizer, device):
  17. messages.append({"role": "user", "content": prompt})
  18. text = tokenizer.apply_chat_template(
  19. messages,
  20. tokenize=False,
  21. add_generation_prompt=True
  22. )
  23. model_inputs = tokenizer([text], return_tensors="pt").to(device)
  24. generate_params = dict(
  25. inputs=model_inputs.input_ids,
  26. max_new_tokens=512,
  27. do_sample=True,
  28. top_k=50,
  29. temperature=0.7,
  30. pad_token_id=tokenizer.eos_token_id,
  31. streamer=streamer
  32. )
  33. thread = Thread(target=model.generate, kwargs=generate_params)
  34. thread.start()
  35. generated_text = ""
  36. for new_text in streamer:
  37. generated_text += new_text
  38. print(new_text, end='', flush=True)
  39. print()
  40. # generated_text
  41. messages.append({"role": "user", "content": generated_text})
  42. In [3]:
  43. # 多轮对话
  44. while True:
  45. user_input = input("User: ")
  46. if user_input.lower() == 'exit':
  47. print("Exiting...")
  48. break
  49. # 生成回复并流式输出
  50. print("Assistant: ", end="")
  51. stream_generate(user_input, model, tokenizer, device)

该代码的目的是使用一个预训练的因果语言模型(在这里是

  1. /gemini/pretrain/Qwen2-0.5B-Instruct

)来生成对用户输入的响应。代码设计为一个多轮对话系统,其中用户可以输入文本,系统(作为“有帮助的助手”)会生成响应。

2.6使用streamlit框架构建即时问答系统

  1. import streamlit as st
  2. from threading import Thread
  3. import json
  4. import time
  5. from transformers import AutoTokenizer
  6. from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
  7. import torch
  8. st.title('Qwen2-0.5B-Instruct')
  9. device = "cuda"
  10. model_name = "/gemini/pretrain/Qwen2-0.5B-Instruct"
  11. @st.cache_resource
  12. def get_model():
  13. model = AutoModelForCausalLM.from_pretrained(model_name).to(device)
  14. tokenizer = AutoTokenizer.from_pretrained(model_name)
  15. return model, tokenizer
  16. model, tokenizer = get_model()
  17. streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
  18. def generate_response(input_text):
  19. prompt = f'''
  20. 你是一个智能问答助手,这是你需要回答的问题:
  21. <question>
  22. {input_text}
  23. </question>
  24. '''
  25. print(prompt)
  26. # 构建消息列表
  27. messages = [{"role": "user", "content": prompt}]
  28. text = tokenizer.apply_chat_template(
  29. messages,
  30. tokenize=False,
  31. add_generation_prompt=True
  32. )
  33. model_inputs = tokenizer([text], return_tensors="pt").to(device)
  34. generate_params = dict(
  35. inputs=model_inputs.input_ids,
  36. max_new_tokens=512,
  37. do_sample=True,
  38. top_k=50,
  39. temperature=0.7,
  40. pad_token_id=tokenizer.eos_token_id,
  41. streamer=streamer
  42. )
  43. thread = Thread(target=model.generate, kwargs=generate_params)
  44. thread.start()
  45. return streamer
  46. with st.form('my_form'):
  47. text = st.text_area('Enter text:', 'What are the three key pieces of advice for learning how to code?')
  48. submitted = st.form_submit_button('Submit')
  49. if submitted:
  50. message_placeholder = st.empty()
  51. # 调用模型生成响应
  52. streamer = generate_response(text)
  53. response = ""
  54. for text in streamer:
  55. response += text
  56. message_placeholder.info(response + "▌")
  57. message_placeholder.info(response)

Streamlit 是一个开源的 Python 库,用于快速创建和共享数据应用。它允许您使用 Python 编写应用逻辑,并自动将结果渲染为网页。在您的代码中,

  1. import streamlit as st

允许您使用 Streamlit 的各种功能,如创建表单、显示文本、图表等。

  1. transformers

库由 Hugging Face 提供,它包含了大量预训练的模型,这些模型可以用于各种自然语言处理任务,如文本生成、分类、命名实体识别等

使用

  1. AutoModelForCausalLM

  1. AutoTokenizer

来加载一个预训练的因果语言模型(通常用于文本生成)和相应的分词器。

三、Stable Diffusion相关准备

3.1环境准备

与千问二环境准备操作相似

创建项目,选择B1.medium规格服务器

选择CUDA12.1 PyTorch2.1.1的镜像

在模型中搜索Qwen2-0.5B-Instruct

点击启动后,即可进入环境。

3.2执行代码

  1. tar xf /gemini/data-1/stable-diffusion-webui.tar -C /gemini/code/
  2. chmod +x /root/miniconda3/lib/python3.10/site-packages/gradio/frpc_linux_amd64_v0.2
  3. cd /stable-diffusion-webui && python launch.py --deepdanbooru --share --xformers --listen

解压 Stable Diffusion WebU

  1. tar xf /gemini/data-1/stable-diffusion-webui.tar -C /gemini/code/

这个命令将

  1. /gemini/data-1/stable-diffusion-webui.tar

文件解压到

  1. /gemini/code/

目录下。

  1. tar xf

用于解压 tar 文件,

  1. -C

选项指定解压的目标目录。

修改 Gradio 相关的文件权限

  1. chmod +x /root/miniconda3/lib/python3.10/site-packages/gradio/frpc_linux_amd64_v0.2
  1. Gradio 库中的一个可执行文件设置执行权限。Gradio 是一个用于快速创建机器学习和深度学习Web应用的Python库。这个命令假设您已经通过某种方式(可能是 Miniconda)安装了 Gradio,并且希望使该二进制文件可执行。请注意,文件路径
  2. **运行 Stable Diffusion WebUI**
  1. cd /stable-diffusion-webui && python launch.py --deepdanbooru --share --xformers --listen

这个命令首先切换到

  1. /stable-diffusion-webui

目录,然后执行

  1. launch.py

脚本来启动 Stable Diffusion WebUI。

  1. --deepdanbooru

  1. --share

  1. --xformers

是传递给

  1. launch.py

的参数,分别可能用于启用 DeepDanbooru 数据集、启用分享功能、以及使用 xformers 库(一个优化后的 transformers 库,用于加速处理)。然而,

  1. --listen

参数后面缺少了一个必要的端口号或监听地址。通常,您会想要指定一个端口号,如

  1. --listen 0.0.0.0:7860

来让 WebUI 在所有接口上的 7860 端口上监听。


本文转载自: https://blog.csdn.net/huodaqi666/article/details/141001531
版权归原作者 huodaqi666 所有, 如有侵权,请联系我们删除。

“【“第二课堂”AI实践课堂笔记】:大模型学习”的评论:

还没有评论