TMElyralab / MusePose

MusePose: a Pose-Driven Image-to-Video Framework for Virtual Human Generation
Other
1.87k stars 125 forks source link

CUDA out of memory on 8 GB VRAM + 8 GB shared RAM #7

Closed nitinmukesh closed 1 month ago

nitinmukesh commented 1 month ago

I installed it on Windows 11 but getting the CUDA out of memory error. I reduced the dance video dimensions to half 540 x 960 still the same.

Anyway to make it work by tweaking any settings like batch size etc....

BlacknoSheep commented 1 month ago

Try

python test_stage_2.py --config ./configs/test_stage_2.yaml -W 200 -H 350
TZYSJTU commented 1 month ago

Please refer to https://github.com/TMElyralab/MusePose/issues/6#issuecomment-2136933975

nitinmukesh commented 1 month ago

Try

python test_stage_2.py --config ./configs/test_stage_2.yaml -W 200 -H 350

Thank you. Following worked for me python test_stage_2.py --config ./configs/test_stage_2.yaml -W 360 -H 640