Jae-Won Chung commited on
Commit
6393815
1 Parent(s): 9d117bb

Fix system prompt setting; FastChat breaking change

Browse files
Files changed (1) hide show
  1. spitfight/prompt.py +1 -6
spitfight/prompt.py CHANGED
@@ -52,12 +52,7 @@ def apply_model_characteristics(
52
  """Apply and return model-specific differences."""
53
  conv = get_conversation_template(model_name)
54
 
55
- if "llama-2" in model_name.lower():
56
- conv.system = f"<s>[INST] <<SYS>>\n{system_prompt}\n<</SYS>>\n\n"
57
- elif "stablelm" in model_name.lower():
58
- conv.system = f"""<|SYSTEM|># {system_prompt}\n"""
59
- else:
60
- conv.system = system_prompt
61
  conv.messages = []
62
  conv.offset = 0
63
 
 
52
  """Apply and return model-specific differences."""
53
  conv = get_conversation_template(model_name)
54
 
55
+ conv.system_message = system_prompt
 
 
 
 
 
56
  conv.messages = []
57
  conv.offset = 0
58