Add support for torch v2.0.0 (#1713)

Fix missing torch._six namespace
This commit is contained in:
Jonáš Kulhánek authored and GitHub committed 2023-04-12 19:17:43 +02:00
1 parent b85cb9f979
commit fd6cc84808
1 file changed
+1 -2
+1 -2
View File
@@ -23,7 +23,6 @@ from typing import Callable, Dict, Union
import torch
import torch.utils.data
from torch._six import string_classes
from nerfstudio.cameras.cameras import Cameras
@@ -119,7 +118,7 @@ def nerfstudio_collate(
return torch.tensor(batch, dtype=torch.float64)
elif isinstance(elem, int):
return torch.tensor(batch)
elif isinstance(elem, string_classes):
elif isinstance(elem, (str, bytes)):
return batch
elif isinstance(elem, collections.abc.Mapping):
try: