From 680d446e4674ff51d1272e7c5705d07d386a1f2c Mon Sep 17 00:00:00 2001
From: lucidrains <lucidrains@gmail.com>
Date: Tue, 3 Oct 2023 09:26:02 -0700
Subject: [PATCH] document in readme later

---
 vit_pytorch/simple_vit_with_register_tokens.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/vit_pytorch/simple_vit_with_register_tokens.py b/vit_pytorch/simple_vit_with_register_tokens.py
index 6b5ab44..774ee85 100644
--- 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