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 |
]
|