ds-nat-j
commented
1 year ago Also I think it's worth investing time adding code that converts the original tensorflow based stylegan2 pre-trained models from .pkl files to PyTorch .pt files within this repo itself.
This can be done using https://github.com/rosinality/stylegan2-pytorch
I've seen other stylegan2 models used in the original demo like stylegan2_human_v2_512.pkl, stylegan2-old-car-config-f.pt etc. If we could parse the model path for these newer models that might come in the future, that would be really awesome.
It would also be great to know how to load these newer models in the code for the time being. Thanks!