mgoin commited on
Commit
261e43f
1 Parent(s): 6a8055c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -7,7 +7,7 @@ tags:
7
  Chat-aligned MPT 7b model pruned to 50% and quantized for inference with DeepSparse
8
 
9
  ```python
10
- import deepsparse
11
- model = deepsparse.TextGeneration(model="hf:mgoin/mpt-7b-chat-50pruned-quant")
12
  model("Tell me a joke.", max_new_tokens=50)
13
  ```
 
7
  Chat-aligned MPT 7b model pruned to 50% and quantized for inference with DeepSparse
8
 
9
  ```python
10
+ from deepsparse import TextGeneration
11
+ model = TextGeneration(model="hf:neuralmagic/mpt-7b-chat-pruned50-quant")
12
  model("Tell me a joke.", max_new_tokens=50)
13
  ```