hpcaitech / Open-Sora

Open-Sora: Democratizing Efficient Video Production for All

Home Page:https://hpcaitech.github.io/Open-Sora/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

python gradio/app.py --port 8001 --host 0.0.0.0 --enable-optimization --model-type v1-HQ-16x512x512 一直报torch.cuda.OutOfMemoryError: CUDA out of memory

yangjianchun opened this issue · comments

显卡A100 40G
python gradio/app.py --port 8001 --host 0.0.0.0 --enable-optimization --model-type v1-HQ-16x512x512 执行后出现torch.cuda.OutOfMemoryError: CUDA out of memory,

但是在命令行下执行CUDA_VISIBLE_DEVICES=1 torchrun --standalone --nproc_per_node 1 scripts/inference.py configs/opensora/inference/16x512x512.py 又没问题,python gradio/app.py 是哪里配置问题吗

You may try replacing the Boolean values from this line and this line with args.enable_optimization.

the
还是一样,还是显示显存溢出,有没还有其他优化方式?

This issue is stale because it has been open for 7 days with no activity.