pandora-s commited on
Commit
3990259
1 Parent(s): b0693ea

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -44,7 +44,7 @@ mistral-chat $HOME/mistral_models/7B-Instruct-v0.3 --instruct --max_tokens 256
44
  ### Instruct following
45
 
46
  ```py
47
- from mistral_inference.model import Transformer
48
  from mistral_inference.generate import generate
49
 
50
  from mistral_common.tokens.tokenizers.mistral import MistralTokenizer
@@ -69,7 +69,7 @@ print(result)
69
 
70
  ```py
71
  from mistral_common.protocol.instruct.tool_calls import Function, Tool
72
- from mistral_inference.model import Transformer
73
  from mistral_inference.generate import generate
74
 
75
  from mistral_common.tokens.tokenizers.mistral import MistralTokenizer
 
44
  ### Instruct following
45
 
46
  ```py
47
+ from mistral_inference.transformer import Transformer
48
  from mistral_inference.generate import generate
49
 
50
  from mistral_common.tokens.tokenizers.mistral import MistralTokenizer
 
69
 
70
  ```py
71
  from mistral_common.protocol.instruct.tool_calls import Function, Tool
72
+ from mistral_inference.transformer import Transformer
73
  from mistral_inference.generate import generate
74
 
75
  from mistral_common.tokens.tokenizers.mistral import MistralTokenizer