VToonify/model/stylegan/op_cpu at main · williamyang1991/VToonify · GitHub

Name		Name	Last commit message	Last commit date
parent directory ..
__init__.py		__init__.py
conv2d_gradfix.py		conv2d_gradfix.py
fused_act.py		fused_act.py
readme.md		readme.md
upfirdn2d.py		upfirdn2d.py

readme.md

Code from rosinality-stylegan2-pytorch-cp

Scripts to convert rosinality/stylegan2-pytorch to the CPU compatible format

If you would like to use CPU for testing or have a problem regarding the cpp extention (fused and upfirdn2d), please make the following changes:

Change model.stylegan.op to model.stylegan.op_cpu

VToonify/util.py

Line 14 in 01b383e

from model.stylegan.op import conv2d_gradfix

VToonify/model/simple_augment.py

Line 12 in 01b383e

from model.stylegan.op import upfirdn2d

VToonify/model/stylegan/model.py

Line 11 in 01b383e

    
           from model.stylegan.op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d, conv2d_gradfix