Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

What is the Memory Requirement? Will 10gb be enough? #17

Open
oisilener1982 opened this issue Jun 1, 2024 · 2 comments
Open

What is the Memory Requirement? Will 10gb be enough? #17

oisilener1982 opened this issue Jun 1, 2024 · 2 comments

Comments

@oisilener1982
Copy link

oisilener1982 commented Jun 1, 2024

Will my RTX 3080 with just 10gb of memory work? Thanks in advance

@tiankuan93
Copy link
Collaborator

In a V100 machine, the following script requires 9873M GPU memory. Please feel free to have a try. Or you can use a shorter audio. For an audio of about 2 seconds, that's roughly 9253M.

python inference.py \
    --reference_image_path "./test_samples/short_case/AOC/ref.jpg" \
    --audio_path "./test_samples/short_case/AOC/aud.mp3" \
    --kps_path "./test_samples/short_case/AOC/kps.pth" \
    --output_path "./output/short_case/talk_AOC_no_retarget.mp4" \
    --retarget_strategy "no_retarget" \
    --num_inference_steps 25

@nitinmukesh
Copy link

Will my RTX 3080 with just 10gb of memory work? Thanks in advance

I tested on 8 GB VRAM (RTX 4060), works absolutely fine

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants