Hugging Face's logo

mjschock
/
mamba-130m

Feature Extraction
Transformers
PyTorch
Safetensors
mamba
custom_code
text-generation-inference
Inference Endpoints
Model card Files Community
mjschock commited on
Commit
5df6816
·
verified ·
1 Parent(s): e38e513

Upload config

Browse files
Files changed (1) hide show
  1. configuration_mamba.py +1 -1
configuration_mamba.py CHANGED
@@ -1,5 +1,5 @@
1
  import mamba_ssm
2
- from transformers import AutoConfig, AutoModel, AutoModelForCausalLM, AutoTokenizer, PretrainedConfig, PreTrainedModel
3
 
4
  mamba_config_defaults = mamba_ssm.models.config_mamba.MambaConfig()
5
 
 
1
  import mamba_ssm
2
+ from transformers import PretrainedConfig
3
 
4
  mamba_config_defaults = mamba_ssm.models.config_mamba.MambaConfig()
5