Update README.md
Browse filesfixing copy/paste issue
README.md
CHANGED
|
@@ -31,10 +31,10 @@ _coming soon_
|
|
| 31 |
## Usage with 🤗 `optimum-neuron`
|
| 32 |
|
| 33 |
```python
|
| 34 |
-
|
| 35 |
|
| 36 |
-
|
| 37 |
-
|
| 38 |
do_sample=True,
|
| 39 |
top_k=10,
|
| 40 |
temperature=0.1,
|
|
|
|
| 31 |
## Usage with 🤗 `optimum-neuron`
|
| 32 |
|
| 33 |
```python
|
| 34 |
+
from optimum.neuron import pipeline
|
| 35 |
|
| 36 |
+
p = pipeline('text-generation', 'jburtoft/CodeLlama-7b-hf-neuron-8xlarge')
|
| 37 |
+
p("import socket\n\ndef ping_exponential_backoff(host: str):",
|
| 38 |
do_sample=True,
|
| 39 |
top_k=10,
|
| 40 |
temperature=0.1,
|