Colab Notebook: Cannot import name 'container_abcs' from 'torch._six'

I’m trying to run the deit colab notebook found here:

https://colab.research.google.com/github/facebookresearch/deit/blob/colab/notebooks/deit_inference.ipynb

but I’m running into an issue in the second cell, specifically the import timm line, which returns this:

ImportError: cannot import name ‘container_abcs’ from ‘torch._six’

>Solution :

 This problem has been previously raised here:

you need to have the last version of timm library, so replace the following line:

!pip install timm==0.3.2

by

!pip install timm