ali-vilab / UniAnimate

Code for Paper "UniAnimate: Taming Unified Video Diffusion Models for Consistent Human Image Animation".
https://unianimate.github.io/
524 stars 29 forks source link

In my docker container,UniAnimate_infer_long will be killed,Why? #35

Open Pythonpa opened 5 days ago

Pythonpa commented 5 days ago

作者您好,我做了一个docker镜像,在容器运行时,能正常执行'python inference.py --cfg configs/UniAnimate_infer.yaml'并得到最终的驱动转化视频。 image

但是,如果运行'python inference.py --cfg configs/UniAnimate_infer_long.yaml'这个长视频的时候,在加载完模型之后,会直接被kill掉。请问是什么原因?触碰到内存瓶颈了? 两个yaml配置中的分辨率都是[512, 768],使用同一张参考图片和同一段动作序列。 我的显卡是3090 24G Hello, I have made a docker image. When the container is running, I can normally execute 'python inference.py -- cfg configs/UniAnimate_infer.yaml ',and get the final driver conversion video. However, if I run the long yaml file 'python inference.py -- cfg configs/UniAnimate_infer_long.yaml ', it will be killed directly after the model is loaded. What is the reason? Did I touch the memory bottleneck? The resolution in both yaml configurations is [512, 768], using the same reference picture and the same motion sequence. My VRAM is 24G 3090

And my base image is from Ubuntu 22.04 CUDA=12.1 Pytorch=2.2.2

123

image

wangxiang1230 commented 5 days ago

Hi, maybe this can be attributed to the fact that your mem is not enough. You can use htop to monitor the used mem.

Pythonpa commented 4 days ago

Hi, maybe this can be attributed to the fact that your mem is not enough. You can use htop to monitor the used mem.

got it,will have another try