Merge pull request #421 from aboutmedicine/master

Update collections.abc import to avoid pytorch 1.8 issue. Fix #420
huggingface · Feb 10, 2021 · b9bd960 · b9bd960 · moghadas76 · Jul 2, 2022
2 parents db6128d + 94ca140
commit b9bd960
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/timm/models/layers/helpers.py b/timm/models/layers/helpers.py
@@ -3,13 +3,13 @@
 Hacked together by / Copyright 2020 Ross Wightman
 """
 from itertools import repeat
-from torch._six import container_abcs
+import collections.abc
 
 
 # From PyTorch internals
 def _ntuple(n):
     def parse(x):
-        if isinstance(x, container_abcs.Iterable):
+        if isinstance(x, collections.abc.Iterable):
             return x
         return tuple(repeat(x, n))
     return parse