Beibars003 commited on
Commit
fc1ded9
·
verified ·
1 Parent(s): 2af54c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -34,9 +34,10 @@ hf_hub_download(
34
 
35
  # Define the prompt markers for Gemma 3
36
  gemma_3_prompt_markers = {
37
- Roles.system: PromptMarkers("", "\n"), # System prompt should be included within user message
38
  Roles.user: PromptMarkers("<start_of_turn>user\n", "<end_of_turn>\n"),
39
- Roles.assistant: PromptMarkers("<start_of_turn>model\n", "<end_of_turn>\n"),
 
40
  Roles.tool: PromptMarkers("", ""), # If you need tool support
41
  }
42
 
 
34
 
35
  # Define the prompt markers for Gemma 3
36
  gemma_3_prompt_markers = {
37
+ Roles.system: PromptMarkers("<start_of_turn>system\n", "<end_of_turn>\n"), # System prompt should be included within user message
38
  Roles.user: PromptMarkers("<start_of_turn>user\n", "<end_of_turn>\n"),
39
+ Roles.assistant: PromptMarkers("<start_of_turn>assistant", ""),
40
+
41
  Roles.tool: PromptMarkers("", ""), # If you need tool support
42
  }
43