Hunyuan-DiT环境搭建&推理测试
引子
最近鹅厂竟然开源了一个多模态的大模型,之前分享福报厂的多模态视觉大模型(
一、环境安装
docker run -it --rm --gpus=all -v /datas/work/zzq:/workspace pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime bash
cd /workspace/HunyuanDiT/HunyuanDiT-main
pip install -r requirements.txt -i
模型下载地址
二、推理测试
改为实际路径,防止下载模型
执行
sh test.sh