document in readme later

lucidrains · Oct 3, 2023 · 680d446 · 680d446
1 parent 3fdb8dd
commit 680d446
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/vit_pytorch/simple_vit_with_register_tokens.py b/vit_pytorch/simple_vit_with_register_tokens.py
@@ -1,3 +1,8 @@
+"""
+    Vision Transformers Need Registers
+    https://arxiv.org/abs/2309.16588
+"""
+
 import torch
 from torch import nn