pytti-tools / pytti-core

https://pytti-tools.github.io/pytti-book/intro.html
MIT License
79 stars 23 forks source link

ModuleNotFoundError: No module named 'torch._six' #227

Closed Sequential-circuits closed 7 months ago

Sequential-circuits commented 9 months ago

ModuleNotFoundError: No module named 'torch._six'

That module I understand is old as hell as it is for python 2, so is the system asking for it?

dmarx commented 7 months ago

you were on the right track but a little off. not that this is a submodule on torch: the issue here is a conflict between the pytorch version in colab and old code in taming-transformers (the code that runs the vqgan). i tweaked the taming fork pytti uses to resolve this issue and it seems to be working, but i think there are certain environment configurations in colab that may still cause other issues. take her for a spin though and let me know what happens