Created a venv, using python 3.10 on Ubuntu 22.04 LTS

Then, used Installation and Initialization steps from the project.

Initialization step error:

(rt2) student@nuc21:~/rt2$ python test.py
Traceback (most recent call last):
  File "/mnt/LexLinux/student/rt2/test.py", line 2, in <module>
    from rt2.model import RT2
  File "/mnt/LexLinux/student/rt2/lib/python3.10/site-packages/rt2/__init__.py", line 1, in <module>
    from rt2.model import RT2
  File "/mnt/LexLinux/student/rt2/lib/python3.10/site-packages/rt2/model.py", line 3, in <module>
    from zeta.structs import (
ImportError: cannot import name 'AutoregressiveWrapper' from 'zeta.structs' (/mnt/LexLinux/student/rt2/lib/python3.10/site-packages/zeta/structs/__init__.py)
(rt2) student@nuc21:~/rt2$

Test app:

import torch
from rt2.model import RT2

# img: (batch_size, 3, 256, 256)
# caption: (batch_size, 1024)
img = torch.randn(1, 3, 256, 256)
caption = torch.randint(0, 20000, (1, 1024))

# model: RT2
model = RT2()

# Run model on img and caption
output = model(img, caption)
print(output)  # (1, 1024, 20000)