DragGAN-main\viz\renderer.py", line 160, in get_network raise NameError('Cannot infer model type from pkl name!') NameError: Cannot infer model type from pkl name!

[babyta]

DragGAN-main\viz\renderer.py", line 160, in get_network raise NameError('Cannot infer model type from pkl name!') NameError: Cannot infer model type from pkl name!

[PDillis]

In viz/renderer.py, it shows that it's expecting the model name to be in the file. That's why in the scripts/download_model.sh script, it first downloads the models and then renames them accordingly. For example, lions_512_pytorch.pkl is renamed to stylegan2_lions_512_pytorch.pkl. So basically, just add stylegan2 if it's a StyleGAN2 model.