No module named 'torch_utils'

mikael-alafriz-deel / lucid-sonic-dreams

MIT License

771 stars 157 forks source link

No module named 'torch_utils' #36

Open shaolinseed opened 2 years ago

shaolinseed commented 2 years ago

Had this working but now getting this error in colab, any suggestions? i've tried installing different versions of pytorch.

shaolinseed commented 2 years ago

it works when i use a default model such as abstract photos , but when i load custom model through drive getting this error

bartman081523 commented 2 years ago

If you have this error, this means your model is a stylegan2 pytorch model and you loading it with the tensorflow version. Or you are loading a stylegan3 model.