* Move weight initialization in constructors. * Fixing mypy for ViT. * remove unnecessary import
Attach a file by drag & drop or click to upload
