colab notebook wrong torch vision version? 2.1 vs. 2.0

[nyck33]

https://colab.research.google.com/github/spmallick/learnopencv/blob/master/PyTorch-Multi-Label-Image-Classification/Multi_Output_Classification.ipynb#scrollTo=TclnDLwbdpbs

getting errors like

TypeError                                 Traceback (most recent call last)

[<ipython-input-17-a3cf0b65dd02>](https://localhost:8080/#) in <cell line: 1>()
----> 1 last_checkpoint_path = train()

[<ipython-input-16-dc3df90b86ee>](https://localhost:8080/#) in train(start_epoch, N_epochs, batch_size, num_workers)
     35         transforms.RandomHorizontalFlip(p=0.5),
     36         transforms.ColorJitter(brightness=0.3, contrast=0.3, saturation=0.3, hue=0),
---> 37         transforms.RandomAffine(degrees=20, translate=(0.1, 0.1), scale=(0.8, 1.2),
     38                                 shear=None, fillcolor=(255, 255, 255)),
     39         transforms.ToTensor(),

TypeError: RandomAffine.__init__() got an unexpected keyword argument 'fillcolor'

The one before it was same error but for resample so I deleted that and got this one.
Discussed here: https://discuss.pytorch.org/t/problem-with-transform-randomaffine-unexpected-argument-fillcolor/26542/3
What is a workaround?