Update README.md
Browse files
README.md
CHANGED
@@ -53,19 +53,11 @@ Read our [blog post](https://zenn.dev/kotoba_tech/articles/f15b2495d44c4f) or ou
|
|
53 |
|
54 |
## Usage
|
55 |
|
56 |
-
First, install additional dependencies in [requirements.txt](./requirements.txt):
|
57 |
-
|
58 |
-
```sh
|
59 |
-
pip install -r requirements.txt
|
60 |
-
```
|
61 |
-
|
62 |
-
### Use the base model
|
63 |
-
|
64 |
git clone [https://github.com/kotoba-tech/kotomamba](https://github.com/kotoba-tech/kotomamba) and follow the repository's README installation section.
|
65 |
|
66 |
-
**WARNING**: huggingface transformers `AutoModelForCausalLM` **doesn't support** mamba model. So, please use
|
67 |
|
68 |
-
You can find the inference sample script in
|
69 |
|
70 |
## Training Datasets
|
71 |
|
|
|
53 |
|
54 |
## Usage
|
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](https://github.com/kotoba-tech/kotomamba/blob/main/scripts/abci/inference/inference_sample.sh)
|
61 |
|
62 |
## Training Datasets
|
63 |
|