Intialization sample error - ImportError: cannot import name 'AutoregressiveWrapper' from 'zeta.structs'

[js333031]

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)

Upvote & Fund

• We're using Polar.sh so you can upvote and help fund this issue.
• We receive the funding once the issue is completed & confirmed by you.
• Thank you in advance for helping prioritize & fund our backlog.

[Eric-Thu]

I found the 'AutoregressiveWrapper' class in zeta is renamed 'AutoRegressiveWrapper' now. So try to replace 'AutoregressiveWrapper' by 'AutoRegressiveWrapper' in model.py. It works fine for me.

[js333031]

Thanks, that worked