how to prune ViT-S from timm library

[sravanthOppo27]

ViT-S is not there in torchvision.models.vision_transformer , It is there in timm.models.vision_transformer . When I was trying to prune the vit_small_patch16_224 , I am getting error as

local_norm = local_norm[idxs] IndexError: index 384 is out of bounds for dimension 0 with size 384 .

Can your method able to prune ViT-S from timm ?

[AymenBOUGUERRA]

@sravanthOppo27 It seems like the author have implemented your request in the latest release https://github.com/VainF/Torch-Pruning/releases/tag/v1.2.1