Update README.md
Browse files
README.md
CHANGED
@@ -57,7 +57,7 @@ git clone [https://github.com/kotoba-tech/kotomamba](https://github.com/kotoba-t
|
|
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
|
61 |
|
62 |
## Training Datasets
|
63 |
|
|
|
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](https://github.com/kotoba-tech/kotomamba/blob/main/scripts/abci/inference/inference_sample.sh)
|
61 |
|
62 |
## Training Datasets
|
63 |
|