VideoCrafter模型部署教程

时间:2024-11-21 19:09:52

一、介绍

VideoCrafter是一个功能强大的AI视频编辑和生成工具,它结合了深度学习和机器学习技术,为用户提供了便捷的视频制作和编辑体验。

系统:Ubuntu22.04系统,显卡:4090,显存:24G

二、基础环境

1.查看系统是否有Miniconda3的虚拟环境

conda -V

如果输入命令没有显示Conda版本号,则需要安装。

安装教程可查看:

屏幕截图

2.更新系统命令

输入下列命令将系统更新及系统缺失命令下载

apt-get update 
apt-get upgrade 
apt-get install -y vim wget unzip lsof net-tools openssh-server git git-lfs gcc cmake build-essential

3.创建虚拟Python环境

  • 创建一个名为"videocrafter"的虚拟镜像,python版本为3.8.5
conda create -n videocrafter python=3.8.5

微信截图_20240909165501.png

  • 进入"videocrafter"虚拟环境
conda activate videocrafter

微信截图_20240909165633.png

4.下载模型

输入下列命令对videocrafter模型进行下载

git clone https://gitclone.com/github.com/AILab-CVC/VideoCrafter.gitls
cd VideoCrafter

微信截图_20240909170134.png

5.下载模型依赖包

输入下列命令:

pip install -r requirements.txt  -i https://pypi.tuna.tsinghua.edu.cn/simple

微信截图_20240913153448.png

出现报错:

微信截图_20240913154403.png

进入requirements.txt文档中将pytorch_lightning的版本号删除

微信截图_20240913154447.png

重新运行下载命令:

问题解决

微信截图_20240913153919.png

耐心等待直到”Successfully“出现: 

微信截图_20240913173213.png

6.添加模型文件

运行下列命令:

sh scripts/run_text2video.sh

微信截图_20240913174950.png

按照错误提示安装checkpoints/base_1024_v1/model.ckpt

创建checkpoints文件夹

mkdir checkpoints
cd checkpoints

微信截图_20240913175351.png

下载模型文件:

git lfs install
git clone https://hf-mirror.com/VideoCrafter/Text2Video-1024

微信截图_20240913180129.png

三、web界面展示

输入下列命令启动界面:

python gradio_app.py

微信截图_20240910095553.png

复制网址打开页面

微信截图_20240910095603.png