Merge pull request huggingface#421 from aboutmedicine/master

Update collections.abc import to avoid pytorch 1.8 issue. Fix huggingface#420
guoriyue · Feb 10, 2021 · 323e7fb · 323e7fb
2 parents c0cf7f6 + 5cdb7af
commit 323e7fb
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