Update model.py

[skr3178]

Correction to model.py:
AutoregressiveWrapper--> AutoRegressiveWrapper

pip list
Package Version

---

accelerate 0.28.0
aiohappyeyeballs 2.4.4
aiohttp 3.11.11
aiosignal 1.3.2
ale-py 0.8.1
annotated-types 0.7.0
async-timeout 5.0.1
atari-py 0.2.5
attrs 24.3.0
AutoROM 0.6.1
AutoROM.accept-rom-license 0.6.1
beartype 0.17.2
bitsandbytes 0.43.0
certifi 2022.12.7
charset-normalizer 2.1.1
classifier-free-guidance-pytorch 0.7.1
click 8.1.7
cloudpickle 3.1.0
cmake 3.25.0
CoLT5-attention 0.10.19
contourpy 1.3.0
cycler 0.12.1
datasets 2.19.2
dill 0.3.8
docker-pycreds 0.4.0
einops 0.7.0
einops-exts 0.0.4
einx 0.3.0
environs 11.2.1
eval_type_backport 0.2.2
exceptiongroup 1.2.2
fairscale 0.4.13
Farama-Notifications 0.0.4
filelock 3.13.1
flake8 7.1.1
fonttools 4.55.1
frozendict 2.4.6
frozenlist 1.5.0
fsspec 2024.3.1
ftfy 6.3.1
gitdb 4.0.11
GitPython 3.1.43
gym 0.26.2
gym-notices 0.0.8
gymnasium 1.0.0
huggingface-hub 0.27.0
idna 3.4
imageio 2.36.1
imageio-ffmpeg 0.5.1
importlib_metadata 8.5.0
importlib_resources 6.4.5
iniconfig 2.0.0
Jinja2 3.1.3
kiwisolver 1.4.7
libcst 1.5.1
lion-pytorch 0.2.3
lit 15.0.7
local-attention 1.9.3
loguru 0.7.3
markdown-it-py 3.0.0
MarkupSafe 2.1.5
marshmallow 3.23.2
matplotlib 3.9.3
mccabe 0.7.0
mdurl 0.1.2
mpmath 1.3.0
multiagent 0.0.1
multidict 6.1.0
multiprocess 0.70.16
networkx 3.2.1
numpy 1.25.2
nvidia-cublas-cu12 12.1.3.1
nvidia-cuda-cupti-cu12 12.1.105
nvidia-cuda-nvrtc-cu12 12.1.105
nvidia-cuda-runtime-cu12 12.1.105
nvidia-cudnn-cu12 8.9.2.26
nvidia-cufft-cu12 11.0.2.54
nvidia-curand-cu12 10.3.2.106
nvidia-cusolver-cu12 11.4.5.107
nvidia-cusparse-cu12 12.1.0.106
nvidia-nccl-cu12 2.19.3
nvidia-nvjitlink-cu12 12.6.85
nvidia-nvtx-cu12 12.1.105
open_clip_torch 2.29.0
opencv-python 4.10.0.84
packaging 24.2
PaLM-rlhf-pytorch 0.2.3
palme 0.1.2
pandas 2.2.3
pillow 10.2.0
pip 24.2
platformdirs 4.3.6
pluggy 1.5.0
propcache 0.2.1
protobuf 5.29.2
psutil 6.1.0
pyarrow 18.1.0
pyarrow-hotfix 0.6
pycodestyle 2.12.1
pydantic 2.10.4
pydantic_core 2.27.2
pyflakes 3.2.0
Pygments 2.18.0
pyparsing 3.2.0
pytest 8.1.1
python-dateutil 2.9.0.post0
python-dotenv 1.0.1
pytz 2024.2
PyYAML 6.0.2
regex 2024.11.6
requests 2.32.3
rich 13.7.1
rt2 0.1.0
safetensors 0.4.5
scipy 1.9.3
sentencepiece 0.2.0
sentry-sdk 2.19.2
setproctitle 1.3.4
setuptools 75.1.0
six 1.16.0
smmap 5.0.1
sympy 1.13.1
tiktoken 0.8.0
timm 1.0.12
tokenizers 0.15.2
tokenmonster 1.1.12
tomli 2.2.1
torch 2.2.0
torchaudio 2.0.1+cu118
torchdiffeq 0.2.5
TorchFix 0.7.0
torchvision 0.17.0
tqdm 4.66.2
transformers 4.39.1
triton 2.2.0
typing 3.7.4.3
typing_extensions 4.12.2
tzdata 2024.2
urllib3 1.26.13
vector_quantize_pytorch 1.14.5
wandb 0.19.1
wcwidth 0.2.13
wheel 0.44.0
xxhash 3.5.0
yarl 1.18.3
zetascale 2.4.5
zipp 3.21.0