Update README.md
Browse files
README.md
CHANGED
|
@@ -24,7 +24,7 @@ IL-7B uses the same prompt format as Zephyr from HF.
|
|
| 24 |
```python
|
| 25 |
import torch
|
| 26 |
from transformers import pipeline
|
| 27 |
-
pipe = pipeline("text-generation", model="cmcmaster/il_7b, torch_dtype=torch.bfloat16, device_map="auto")
|
| 28 |
messages = [
|
| 29 |
{"role": "user", "content": "How many helicopters can a human eat in one sitting?"},
|
| 30 |
]
|
|
|
|
| 24 |
```python
|
| 25 |
import torch
|
| 26 |
from transformers import pipeline
|
| 27 |
+
pipe = pipeline("text-generation", model="cmcmaster/il_7b", torch_dtype=torch.bfloat16, device_map="auto")
|
| 28 |
messages = [
|
| 29 |
{"role": "user", "content": "How many helicopters can a human eat in one sitting?"},
|
| 30 |
]
|