Update README.md

De30 · Apr 15, 2024 · 4853598 · 4853598
1 parent 220aae8
commit 4853598
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/demo/README.md b/demo/README.md
@@ -37,7 +37,7 @@ client = OpenAI(
     base_url=API_URL,
     api_key="EMPTY",
 )
-
+system_prompt = "A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions. "
 stop_tokens = []
 completion = client.chat.completions.create(
     model=model_path,
@@ -46,6 +46,7 @@ completion = client.chat.completions.create(
     max_tokens=4096,
     stop=stop_tokens,
     messages=[
+    {'role': 'system', 'content': system_prompt},
     {"role": "user", "content": "Hello! What is your name?"},
     {"role": "assistant", "content": "I am WizardLM2!"},
     {"role": "user", "content": "Nice to meet you!"},