token-shift-gpt

Depends On

• einops (einops>=0.3)
• torch (torch>=1.6)