diff --git a/imagebind_LLM/util/misc.py b/imagebind_LLM/util/misc.py
index 6bb622e..481796a 100644
--- a/imagebind_LLM/util/misc.py
+++ b/imagebind_LLM/util/misc.py
@@ -21,7 +21,7 @@
 import torch
 import torch.utils.data
 import torch.distributed as dist
-from torch._six import inf
+from torch import inf
 
 
 class SmoothedValue(object):
@@ -410,4 +410,4 @@ def download(url: str, root: str):
                 loop.update(len(buffer))
 
 
-    return download_target
\ No newline at end of file
+    return download_target