cmcmaster commited on
Commit
01c2610
1 Parent(s): 805ed9b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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
  ]