runtime error

Space failed. Exit code: 1. Reason: Traceback (most recent call last): File "app.py", line 10, in <module> from taming.models import vqgan File "/home/user/app/src/taming-transformers/taming/models/vqgan.py", line 5, in <module> from main import instantiate_from_config File "/home/user/app/src/taming-transformers/main.py", line 14, in <module> from taming.data.utils import custom_collate File "/home/user/app/src/taming-transformers/taming/data/utils.py", line 11, in <module> from torch._six import string_classes ModuleNotFoundError: No module named 'torch._six'

Container logs: