From 72e7ee22b1cac50bf9b41ff39d2c1da85d98b656 Mon Sep 17 00:00:00 2001
From: Brett Young <32812705+bdytx5@users.noreply.github.com>
Date: Wed, 9 Aug 2023 17:00:31 -0400
Subject: [PATCH] Update vit_utils.py

add torch 2.0 functionality
---
 timesformer/models/vit_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/timesformer/models/vit_utils.py b/timesformer/models/vit_utils.py
index 9ce6a93..8c799c6 100755
--- a/timesformer/models/vit_utils.py
+++ b/timesformer/models/vit_utils.py
@@ -11,7 +11,7 @@
 from timesformer.models.helpers import load_pretrained
 from .build import MODEL_REGISTRY
 from itertools import repeat
-from torch._six import container_abcs
+import collections.abc as container_abcs
 
 DEFAULT_CROP_PCT = 0.875
 IMAGENET_DEFAULT_MEAN = (0.485, 0.456, 0.406)