have a language model address https://github.com/lucidrains/vit-pytorch/issues/348

2025-12-30 08:02:29 +00:00 · 2025-09-25 06:21:13 -07:00
parent f6bc14c81d
commit 3becf087bb
6 changed files with 71 additions and 51 deletions
--- a/tests/.DS_Store
+++ b/tests/.DS_Store
--- a/tests/test_vit.py
+++ b/tests/test_vit.py
@@ -1,7 +1,7 @@
 import torch
 from vit_pytorch import ViT

-def test():
+def test_vit():
    v = ViT(
        image_size = 256,
        patch_size = 32,