Spaces:
Running
Running
Jae-Won Chung
commited on
Commit
•
6393815
1
Parent(s):
9d117bb
Fix system prompt setting; FastChat breaking change
Browse files- 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 |
-
|
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 |
|