Hugging Face's logo

kotoba-tech
/
kotomamba-2.8B-v1.0

Text Generation
Transformers
PyTorch
English
Japanese
Inference Endpoints
Model card Files Community
1
okoge commited on
Commit
f21ceab
·
verified ·
1 Parent(s): 4a5cd20

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -55,7 +55,7 @@ Read our [blog post](https://zenn.dev/kotoba_tech/articles/f15b2495d44c4f) or ou
55
 
56
  git clone [https://github.com/kotoba-tech/kotomamba](https://github.com/kotoba-tech/kotomamba) and follow the repository's README installation section.
57
 
58
- **WARNING**: huggingface transformers `AutoModelForCausalLM` **doesn't support** mamba model. So, please use `kotomamba/benchmarks/benchmark_generation_mamba_simple.py`
59
 
60
  You can find the inference sample script in `scripts/abci/inference/inference_sample.sh`
61
 
 
55
 
56
  git clone [https://github.com/kotoba-tech/kotomamba](https://github.com/kotoba-tech/kotomamba) and follow the repository's README installation section.
57
 
58
+ **WARNING**: huggingface transformers `AutoModelForCausalLM` **doesn't support** mamba model. So, please use [kotomamba/benchmarks/benchmark_generation_mamba_simple.py](https://github.com/kotoba-tech/kotomamba/blob/main/benchmarks/benchmark_generation_mamba_simple.py)
59
 
60
  You can find the inference sample script in `scripts/abci/inference/inference_sample.sh`
61