【VisualGLM】大模型之 VisualGLM 部署

目录

1. VisualGLM 效果展示

2. VisualGLM 介绍

3. VisualGLM 部署

---

1. VisualGLM 效果展示

• VisualGLM 问答

 

• 原始图片

 

2. VisualGLM 介绍

VisualGLM 主要做的是通过图像生成文字，而 Stable Diffusion 是通过文字生成图像。

 

一种方法是将图像当作一种特殊的语言进行预训练，还有一种是将图像特征对齐到预训练语言模型，可充分利用语言模型，并且无缝衔接多轮对话能力，但提取图像语义特征会损失底层信息。

 

VisualGLM 模型架构是 ViT + QFormer + ChatGLM，在预训练阶段对 QFormer 和 ViT LoRA 进行训练，在微调阶段对 QFormer 和 ChatGLM LoRA 进行训练，训练目标是自回归损失（根据图像生成正确的文本）和对比损失（输入 ChatGLM 的视觉特征与对应文本的语义特征对齐）。

3. VisualGLM 部署

• ChatGLM 模型权重下载见

【ChatGLM】大模型之 ChatGLM 部署

• VisualGLM 启动

# 源码下载
git clone https://github.com/THUDM/VisualGLM-6B.git

# 代码修改
# 可以把 ChatGLM-6B 改为本地加载

# web 端启动
cd VisualGLM-6B
pip install -r requirements.txt
pytohn web_deme.py