From 3b4ca336be45742c852b31e20b6d82be4b32f5fd Mon Sep 17 00:00:00 2001
From: "O. Inha" <50331907+olaviinha@users.noreply.github.com>
Date: Tue, 13 Jun 2023 14:29:03 +0300
Subject: [PATCH] Drop nonexisting module torch._six

Fix `ModuleNotFoundError: No module named 'torch._six'`
---
 taming/data/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/taming/data/utils.py b/taming/data/utils.py
index 2b3c3d53..df5ed249 100644
--- a/taming/data/utils.py
+++ b/taming/data/utils.py
@@ -8,10 +8,10 @@
 import numpy as np
 import torch
 from taming.data.helper_types import Annotation
-from torch._six import string_classes
 from torch.utils.data._utils.collate import np_str_obj_array_pattern, default_collate_err_msg_format
 from tqdm import tqdm
 
+string_classes = str
 
 def unpack(path):
     if path.endswith("tar.gz"):