Update README.md
Browse files
README.md
CHANGED
@@ -82,8 +82,8 @@ model = AutoGPTQForCausalLM.from_quantized(model_name_or_path,
|
|
82 |
|
83 |
# Note: check the prompt template is correct for this model.
|
84 |
prompt = "Tell me about AI"
|
85 |
-
prompt_template=f'''###
|
86 |
-
###
|
87 |
|
88 |
print("\n\n*** Generate:")
|
89 |
|
|
|
82 |
|
83 |
# Note: check the prompt template is correct for this model.
|
84 |
prompt = "Tell me about AI"
|
85 |
+
prompt_template=f'''###Human: {prompt}
|
86 |
+
###Assistant:'''
|
87 |
|
88 |
print("\n\n*** Generate:")
|
89 |
|