Update README.md
Browse files
README.md
CHANGED
|
@@ -37,7 +37,7 @@ tokens = tokenizer.encode_chat_completion(completion_request).tokens
|
|
| 37 |
## Inference with `mistral_inference`
|
| 38 |
|
| 39 |
```py
|
| 40 |
-
from mistral_inference.
|
| 41 |
from mistral_inference.generate import generate
|
| 42 |
|
| 43 |
model = Transformer.from_folder(mistral_models_path)
|
|
|
|
| 37 |
## Inference with `mistral_inference`
|
| 38 |
|
| 39 |
```py
|
| 40 |
+
from mistral_inference.transformer import Transformer
|
| 41 |
from mistral_inference.generate import generate
|
| 42 |
|
| 43 |
model = Transformer.from_folder(mistral_models_path)
|