Update distill.py#318

vivekh2000 wants to merge 1 commit intolucidrains:mainlucidrains/vit-pytorch:mainfrom

vivekh2000:patch-2vivekh2000/vit-pytorch:patch-2Copy head branch name to clipboard