Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[BUG] from zeta.nn import SSM TypeError: code expected at least 16 arguments, got 15 #314

Open
jhkwag970 opened this issue Jan 11, 2025 · 1 comment
Assignees
Labels
bug Something isn't working

Comments

@jhkwag970
Copy link

Describe the bug
pip3 install -U zetascale and using from zeta.nn import SSM causes TypeError: code expected at least 16 arguments, got 15

  File "/research/nfs_yilmaz_15/kwag/test/vim.py", line 7, in <module>
    from zeta.nn import SSM
  File "/home/kwag.3/anaconda3/envs/segformer3d/lib/python3.11/site-packages/zeta/__init__.py", line 12, in <module>
    from zeta.training import *  # noqa: F403, E402
    ^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/kwag.3/anaconda3/envs/segformer3d/lib/python3.11/site-packages/zeta/training/__init__.py", line 2, in <module>
    from zeta.training.dataloader import build_dataloaders, build_pre_tokenized
  File "/home/kwag.3/anaconda3/envs/segformer3d/lib/python3.11/site-packages/zeta/training/dataloader.py", line 3, in <module>
    from datasets import load_dataset
  File "/home/kwag.3/anaconda3/envs/segformer3d/lib/python3.11/site-packages/datasets/__init__.py", line 22, in <module>
    from .arrow_dataset import Dataset
  File "/home/kwag.3/anaconda3/envs/segformer3d/lib/python3.11/site-packages/datasets/arrow_dataset.py", line 58, in <module>
    import pandas as pd
  File "/home/kwag.3/anaconda3/envs/segformer3d/lib/python3.11/site-packages/pandas/__init__.py", line 26, in <module>
    from pandas.compat import (
  File "/home/kwag.3/anaconda3/envs/segformer3d/lib/python3.11/site-packages/pandas/compat/__init__.py", line 27, in <module>
    from pandas.compat.pyarrow import (
  File "/home/kwag.3/anaconda3/envs/segformer3d/lib/python3.11/site-packages/pandas/compat/pyarrow.py", line 8, in <module>
    import pyarrow as pa
  File "/home/kwag.3/anaconda3/envs/segformer3d/lib/python3.11/site-packages/pyarrow/__init__.py", line 65, in <module>
    import pyarrow.lib as _lib
  File "pyarrow/compat.pxi", line 37, in init pyarrow.lib
  File "/home/kwag.3/anaconda3/envs/segformer3d/lib/python3.11/site-packages/cloudpickle/__init__.py", line 3, in <module>
    from cloudpickle.cloudpickle import *
  File "/home/kwag.3/anaconda3/envs/segformer3d/lib/python3.11/site-packages/cloudpickle/cloudpickle.py", line 145, in <module>
    _cell_set_template_code = _make_cell_set_template_code()
                              ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/kwag.3/anaconda3/envs/segformer3d/lib/python3.11/site-packages/cloudpickle/cloudpickle.py", line 126, in _make_cell_set_template_code
    return types.CodeType(
           ^^^^^^^^^^^^^^^
TypeError: code expected at least 16 arguments, got 15
# Name                    Version                   Build  Channel
_libgcc_mutex             0.1                        main  
_openmp_mutex             5.1                       1_gnu  
accelerate                0.27.2                   pypi_0    pypi
aiohappyeyeballs          2.4.4                    pypi_0    pypi
aiohttp                   3.10.11                  pypi_0    pypi
aiosignal                 1.3.1                    pypi_0    pypi
argparse                  1.4.0                    pypi_0    pypi
async-timeout             5.0.1                    pypi_0    pypi
attrs                     24.3.0                   pypi_0    pypi
awscli                    1.36.38                  pypi_0    pypi
backports-zoneinfo        0.2.1                    pypi_0    pypi
beartype                  0.17.2                   pypi_0    pypi
bitsandbytes              0.42.0                   pypi_0    pypi
boto3                     1.35.97                  pypi_0    pypi
botocore                  1.35.97                  pypi_0    pypi
ca-certificates           2024.12.31           h06a4308_0  
cachetools                5.5.0                    pypi_0    pypi
certifi                   2024.12.14               pypi_0    pypi
cffi                      1.17.1                   pypi_0    pypi
charset-normalizer        3.4.1                    pypi_0    pypi
click                     8.1.8                    pypi_0    pypi
colorama                  0.4.4                    pypi_0    pypi
colt5-attention           0.10.20                  pypi_0    pypi
cryptography              44.0.0                   pypi_0    pypi
datasets                  2.19.2                   pypi_0    pypi
dill                      0.3.8                    pypi_0    pypi
docutils                  0.16                     pypi_0    pypi
einops                    0.7.0                    pypi_0    pypi
einops-exts               0.0.4                    pypi_0    pypi
einx                      0.3.0                    pypi_0    pypi
exceptiongroup            1.2.2                    pypi_0    pypi
filelock                  3.16.1                   pypi_0    pypi
flake8                    7.1.1                    pypi_0    pypi
frozendict                2.4.6                    pypi_0    pypi
frozenlist                1.5.0                    pypi_0    pypi
fsspec                    2024.3.1                 pypi_0    pypi
huggingface-hub           0.27.1                   pypi_0    pypi
idna                      3.10                     pypi_0    pypi
importlib-resources       6.4.5                    pypi_0    pypi
iniconfig                 2.0.0                    pypi_0    pypi
jinja2                    3.1.5                    pypi_0    pypi
jmespath                  1.0.1                    pypi_0    pypi
jsonschema                4.23.0                   pypi_0    pypi
jsonschema-specifications 2023.12.1                pypi_0    pypi
ld_impl_linux-64          2.40                 h12ee557_0  
libcst                    1.1.0                    pypi_0    pypi
libffi                    3.4.4                h6a678d5_1  
libgcc-ng                 11.2.0               h1234567_1  
libgomp                   11.2.0               h1234567_1  
libstdcxx-ng              11.2.0               h1234567_1  
lion-pytorch              0.1.2                    pypi_0    pypi
local-attention           1.9.3                    pypi_0    pypi
loguru                    0.7.3                    pypi_0    pypi
markdown-it-py            3.0.0                    pypi_0    pypi
markupsafe                2.1.5                    pypi_0    pypi
mccabe                    0.7.0                    pypi_0    pypi
mdurl                     0.1.2                    pypi_0    pypi
mpmath                    1.3.0                    pypi_0    pypi
multidict                 6.1.0                    pypi_0    pypi
multiprocess              0.70.16                  pypi_0    pypi
mypy-extensions           1.0.0                    pypi_0    pypi
ncurses                   6.4                  h6a678d5_0  
networkx                  3.1                      pypi_0    pypi
numexpr                   2.8.6                    pypi_0    pypi
numpy                     1.24.4                   pypi_0    pypi
nvidia-cublas-cu12        12.1.3.1                 pypi_0    pypi
nvidia-cuda-cupti-cu12    12.1.105                 pypi_0    pypi
nvidia-cuda-nvrtc-cu12    12.1.105                 pypi_0    pypi
nvidia-cuda-runtime-cu12  12.1.105                 pypi_0    pypi
nvidia-cudnn-cu12         8.9.2.26                 pypi_0    pypi
nvidia-cufft-cu12         11.0.2.54                pypi_0    pypi
nvidia-curand-cu12        10.3.2.106               pypi_0    pypi
nvidia-cusolver-cu12      11.4.5.107               pypi_0    pypi
nvidia-cusparse-cu12      12.1.0.106               pypi_0    pypi
nvidia-nccl-cu12          2.19.3                   pypi_0    pypi
nvidia-nvjitlink-cu12     12.6.85                  pypi_0    pypi
nvidia-nvtx-cu12          12.1.105                 pypi_0    pypi
openssl                   3.0.15               h5eee18b_0  
packaging                 24.2                     pypi_0    pypi
pandas                    2.0.3                    pypi_0    pypi
pendulum                  3.0.0                    pypi_0    pypi
pillow                    10.4.0                   pypi_0    pypi
pip                       24.2             py38h06a4308_0  
pkgutil-resolve-name      1.3.10                   pypi_0    pypi
pluggy                    1.5.0                    pypi_0    pypi
prettytable               3.11.0                   pypi_0    pypi
propcache                 0.2.0                    pypi_0    pypi
psutil                    6.1.1                    pypi_0    pypi
pulp                      2.9.0                    pypi_0    pypi
pyarrow                   17.0.0                   pypi_0    pypi
pyarrow-hotfix            0.6                      pypi_0    pypi
pyasn1                    0.6.1                    pypi_0    pypi
pycodestyle               2.12.1                   pypi_0    pypi
pycparser                 2.22                     pypi_0    pypi
pyflakes                  3.2.0                    pypi_0    pypi
pygments                  2.19.1                   pypi_0    pypi
pytest                    8.0.2                    pypi_0    pypi
python                    3.8.20               he870216_0  
python-dateutil           2.9.0.post0              pypi_0    pypi
python-dotenv             1.0.1                    pypi_0    pypi
pytz                      2024.2                   pypi_0    pypi
pyyaml                    6.0.2                    pypi_0    pypi
readline                  8.2                  h5eee18b_0  
referencing               0.35.1                   pypi_0    pypi
regex                     2024.11.6                pypi_0    pypi
requests                  2.32.3                   pypi_0    pypi
rich                      13.7.0                   pypi_0    pypi
rpds-py                   0.20.1                   pypi_0    pypi
rsa                       4.7.2                    pypi_0    pypi
s3transfer                0.10.4                   pypi_0    pypi
safetensors               0.5.2                    pypi_0    pypi
scipy                     1.9.3                    pypi_0    pypi
sentencepiece             0.2.0                    pypi_0    pypi
setuptools                75.1.0           py38h06a4308_0  
six                       1.17.0                   pypi_0    pypi
skypilot                  0.4.1                    pypi_0    pypi
sqlite                    3.45.3               h5eee18b_0  
sympy                     1.13.3                   pypi_0    pypi
tabulate                  0.9.0                    pypi_0    pypi
tiktoken                  0.6.0                    pypi_0    pypi
time-machine              2.15.0                   pypi_0    pypi
timm                      0.9.16                   pypi_0    pypi
tk                        8.6.14               h39e8969_0  
tokenizers                0.15.2                   pypi_0    pypi
tokenmonster              1.1.12                   pypi_0    pypi
tomli                     2.2.1                    pypi_0    pypi
torch                     2.2.0                    pypi_0    pypi
torchaudio                2.4.1                    pypi_0    pypi
torchdiffeq               0.2.3                    pypi_0    pypi
torchfix                  0.3.0                    pypi_0    pypi
torchvision               0.17.0                   pypi_0    pypi
tqdm                      4.66.2                   pypi_0    pypi
transformers              4.38.2                   pypi_0    pypi
triton                    2.2.0                    pypi_0    pypi
typing                    3.7.4.3                  pypi_0    pypi
typing-extensions         4.12.2                   pypi_0    pypi
typing-inspect            0.9.0                    pypi_0    pypi
tzdata                    2024.2                   pypi_0    pypi
urllib3                   1.26.20                  pypi_0    pypi
vector-quantize-pytorch   1.14.1                   pypi_0    pypi
wcwidth                   0.2.13                   pypi_0    pypi
wheel                     0.44.0           py38h06a4308_0  
xxhash                    3.5.0                    pypi_0    pypi
xz                        5.4.6                h5eee18b_1  
yarl                      1.15.2                   pypi_0    pypi
zetascale                 2.2.7                    pypi_0    pypi
zipp                      3.20.2                   pypi_0    pypi
zlib                      1.2.13               h5eee18b_1 

library list

Thank you!

@jhkwag970 jhkwag970 added the bug Something isn't working label Jan 11, 2025
Copy link

Hello there, thank you for opening an Issue ! 🙏🏻 The team was notified and they will get back to you asap.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

No branches or pull requests

2 participants