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