Update README.md
Browse files
README.md
CHANGED
|
@@ -156,6 +156,7 @@ model = AutoGPTQForCausalLM.from_quantized(model_name_or_path,
|
|
| 156 |
"""
|
| 157 |
|
| 158 |
prompt = "Tell me about AI"
|
|
|
|
| 159 |
prompt_template=f'''SYSTEM: {system_message}
|
| 160 |
USER: {prompt}
|
| 161 |
ASSISTANT:
|
|
|
|
| 156 |
"""
|
| 157 |
|
| 158 |
prompt = "Tell me about AI"
|
| 159 |
+
system_message = "You are a helpful assistant"
|
| 160 |
prompt_template=f'''SYSTEM: {system_message}
|
| 161 |
USER: {prompt}
|
| 162 |
ASSISTANT:
|