CodeFormer icon indicating copy to clipboard operation
CodeFormer copied to clipboard

Lower pretrained model resolution possible?

Open instant-high opened this issue 2 years ago • 0 comments

Is it possible to train, or has someone, willing to share, a lower resolution trained model? e.g. 256x256 pixel ? I would need it for speed of inference reason and limited GPU memory. I can run codeformer on my GTX 1050 / 2GB, but it's a bit slow (1.6 it/sec.)

On the same GPU I can run GPEN 256 togehter with faceparsing and thin plate spline motion model combined in one inference script.

Regards Thomas

Edit: The project mentioned above is a rewritten wav2lip using insightface/onnx for face crop and alignment, faceparsing for masking the face, tpsmm for optional animating the or swapping another face, and gpen for final face enhancment. Depending on torch version and graphic driver version, I can run all together in one in inference.

instant-high avatar Aug 17 '23 12:08 instant-high