instant-high / wav2lip-onnx-HQ

Full version of wav2lip-onnx including face alignment and face enhancement and more...
31 stars 7 forks source link

How do I accelerate with wav2lip_batch_size #3

Open chensav opened 2 weeks ago

chensav commented 2 weeks ago

I want to run the gpu to full speed

instant-high commented 2 weeks ago

To keep the code simple I completely removed batch size in this version. (also when I made this I only had GPU 2GB, so I couldn' use larger batch + mask + enhancer)

Nevertheless the converted onnx-models support batch size...

I will not implement it again, but maybe if you want to do it, look at the simple version here: https://github.com/instant-high/wav2lip-onnx Main differences I think are in DATAGEN function.

It supports wav2lip_batch_size parameter. But no face alignment, masking or enhancer