Update README.md
Browse files
README.md
CHANGED
|
@@ -37,6 +37,7 @@ You can use the models through Huggingface's Transformers library. Check our Git
|
|
| 37 |
```python
|
| 38 |
from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
|
| 39 |
import torch
|
|
|
|
| 40 |
from string import Template
|
| 41 |
prompt_template = Template("Human: ${inst} </s> Assistant: ")
|
| 42 |
|
|
|
|
| 37 |
```python
|
| 38 |
from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
|
| 39 |
import torch
|
| 40 |
+
import re
|
| 41 |
from string import Template
|
| 42 |
prompt_template = Template("Human: ${inst} </s> Assistant: ")
|
| 43 |
|