Update README.md
Browse files
README.md
CHANGED
@@ -82,7 +82,6 @@ After installing `mistral_inference`, a `mistral-demo` CLI command should be ava
|
|
82 |
> ```sh
|
83 |
> pip install mistral_inference
|
84 |
> pip install mistral-demo
|
85 |
-
> pip install accelerate #GPU A100/L4
|
86 |
> pip install git+https://github.com/huggingface/transformers.git
|
87 |
> ```
|
88 |
|
@@ -102,6 +101,8 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
|
102 |
## Accelerator mode:
|
103 |
|
104 |
```py
|
|
|
|
|
105 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
106 |
from accelerate import Accelerator
|
107 |
|
|
|
82 |
> ```sh
|
83 |
> pip install mistral_inference
|
84 |
> pip install mistral-demo
|
|
|
85 |
> pip install git+https://github.com/huggingface/transformers.git
|
86 |
> ```
|
87 |
|
|
|
101 |
## Accelerator mode:
|
102 |
|
103 |
```py
|
104 |
+
pip install accelerate #GPU A100/L4
|
105 |
+
|
106 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
107 |
from accelerate import Accelerator
|
108 |
|