skyworkai / gamba Goto Github PK

Thanks a lot for the great work! Yet I encourtered some problems while runing the demo

Need help for the error for the configuration loading

I am trying to run the test based on the code. Yet, I found one error in:

It says the input arg for defaults (config_defaults) should have len(defaults) > 2.

While I am not familiar to tyro, could you guys tell a bit on how to solve this?

Extra dependency for the repo:

pip install timm jaxtyping omegaconf typeguard colorama

Minor change for the mamba_ssm:

For the latest version of mamba_ssm (2.2.2) the following line:

should be changed to from mamba_ssm.ops.triton.layer_norm import RMSNorm, layer_norm_fn, rms_norm_fn

I used python 3.8, but when I 'pip install causal-conv1d==1.2.0 mamba-ssm' , I got the following errors. How can I solve?

ERROR: Could not find a version that satisfies the requirement causal-conv1d==1.2.0 (from versions: 1.0.0, 1.0.1, 1.0.2, 1.1.0, 1.1.1, 1.1.2.post1, 1.1.3.post1, 1.2.0.post1, 1.2.0.post2, 1.2.1, 1.2.2.post1, 1.3.0.post1, 1.4.0)
ERROR: No matching distribution found for causal-conv1d==1.2.0