antgroup / echomimic

EchoMimic: Lifelike Audio-Driven Portrait Animations through Editable Landmark Conditioning
https://antgroup.github.io/ai/echomimic/
Apache License 2.0
3.06k stars 356 forks source link

Refinement for gpu settings #119

Open CodeLyokoscj opened 3 months ago

CodeLyokoscj commented 3 months ago

This PR is to provide a unified interface for gpu settings. One can change the value of "device" in default_values of EchoMimic/webgui.py to modify gpu settings easily.

default_values = { "width": 512, "height": 512, "length": 1200, "seed": 420, "facemask_dilation_ratio": 0.1, "facecrop_dilation_ratio": 0.5, "context_frames": 12, "context_overlap": 3, "cfg": 2.5, "steps": 30, "sample_rate": 16000, "fps": 24, "device": "cuda" }

medalawi commented 3 months ago

how to change batch size or lower the settings?