I want to convert .pth model from StyleGAN v2 (pytorch implementation) to .pt (torchscript model) for triton inference with jit trace. How I could do that ?
How I could convert cuda function to torchscript ? fused_bias_act and upfirdn_2d module ?
I have no documentation anywhere to do that.
Thanks a lot.