Open wangyf8848 opened 2 weeks ago
How to deal with this error: ImportError: cannot import name 'get_full_repo_name' from 'huggingface_hub'
when I run the code below: `import torch from mamba_ssm import Mamba
batch, length, dim = 2, 64, 16 x = torch.randn(batch, length, dim).to("cuda") model = Mamba(
d_model=dim, # Model dimension d_model d_state=16, # SSM state expansion factor d_conv=4, # Local convolution width expand=2, # Block expansion factor
).to("cuda") y = model(x) assert y.shape == x.shape print(y) `
How to deal with this error: ImportError: cannot import name 'get_full_repo_name' from 'huggingface_hub'
when I run the code below: `import torch from mamba_ssm import Mamba
batch, length, dim = 2, 64, 16 x = torch.randn(batch, length, dim).to("cuda") model = Mamba(
This module uses roughly 3 expand d_model^2 parameters
).to("cuda") y = model(x) assert y.shape == x.shape print(y) `