For CPU compatible, missing "import contextlib"

williamyang1991 / VToonify

[SIGGRAPH Asia 2022] VToonify: Controllable High-Resolution Portrait Video Style Transfer

Other

3.54k stars 446 forks source link

For CPU compatible, missing "import contextlib" #9

Closed RicoLeehdu closed 2 years ago

RicoLeehdu commented 2 years ago

The file named 'mode/stylegan/op_cpu/conv2d_gradfix.py' missing "import contextlib" at beginning, which is refered latter as "@contextlib.contextmanager"

williamyang1991 commented 2 years ago

Thank you! I have fixed it.

RicoLeehdu commented 2 years ago

Thank you! I have fixed it.

Thanks for your excellent work !