ubuntu20.04下Stable Video Diffusion安装和测试

github | https://github.com/Stability-AI/generative-models
Hugging Face | https://huggingface.co/stabilityai/stable-video-diffusion-img2vid-xt
Paper | https://stability.ai/research/stable-video-diffusion-scaling-latent-video-diffusion-models-to-large-datasets

下载

git clone https://github.com/Stability-AI/generative-models
cd generative-models

下载模型

SVD | https://huggingface.co/stabilityai/stable-video-diffusion-img2vid-xt
SVD-XT | https://huggingface.co/stabilityai/stable-video-diffusion-img2vid

模型有4个，任意一个都可以使用，存放目录：

generative-models/checkpoints/

python环境配置

conda create --name svd python=3.10 -y
source activate svd
pip3 install -r requirements/pt2.txt
pip3 install .

 运行

cd generative-models
streamlit run scripts/demo/video_sampling.py  --server.address  0.0.0.0  --server.port 7862

 报错：

报错1：如果没有安装ffmpeg，就安装一下

apt install ffmpeg

报错2：

TypeError: TiffWriter.write() got an unexpected keyword argument 'fps'

解决办法：安装

pip install imageio==2.19.3
pip install imageio-ffmpeg==0.4.7