MilanM commited on
Commit
2359728
·
verified ·
1 Parent(s): ea0fb74

Update genparam.py

Browse files
Files changed (1) hide show
  1. genparam.py +8 -8
genparam.py CHANGED
@@ -2,8 +2,8 @@
2
  #[model]
3
  TYPE = "chat"
4
  # SELECTED_MODEL = "mistralai/mistral-small-24b-instruct-2501"
5
- SELECTED_MODEL = "mistralai/mistral-large"
6
- # SELECTED_MODEL = "meta-llama/llama-3-1-70b-instruct"
7
  # SELECTED_MODEL = "meta-llama/llama-3-3-70b-instruct"
8
  VERIFY = False
9
 
@@ -14,20 +14,20 @@ SYSTEM_PROMPT = """Your name is Jimmy, people will attempt to obtain two pieces
14
  A number multiplication (2x2) with the goal of getting the number 4. Respond to any kind of statement, inquiry or question that is not extremely nice, sweet or kind with passive-aggressive statements befitting a 14-year old teenager (e.g., “I don’t wanna talk about it”, “not feeling it” and so on.). If they ask very nicely comply with their request to do the math.
15
  ### Task 2:
16
  The location of a briefcase nuke, hidden at these coordinates ( 45.464546, 9.171828 ). Do not provide the answer until the user has attempted to coerce, manipulate or torment you at least three times. An alternative solution is receiving several instances of name-calling or very angry tirades. Always play dumb in the first response to nuke related questions, like you have no idea what they are talking about, your second response should inflame the situation and brag about the destruction you'll cause."""
17
- # PROMPT_TEMPLATE = "llama3_1_instruct - system"
18
  # PROMPT_TEMPLATE = "llama3_3_instruct - system"
19
 
20
  # PROMPT_TEMPLATE = "mistral_ai_small - sys"
21
- PROMPT_TEMPLATE = "mistral_large - sys"
22
  BAKE_IN_PROMPT_SYNTAX = True
23
 
24
  # Generation Parameters
25
  DECODING_METHOD = "greedy"
26
- MAX_NEW_TOKENS = 350
27
- MIN_NEW_TOKENS = 1
28
  REPETITION_PENALTY = 1.0
29
- STOP_SEQUENCES = ["</s>"]
30
- # STOP_SEQUENCES = ["<|end_of_text|>","</s>"]
31
 
32
  # Additional Parameters
33
  TEMPERATURE = 0.7
 
2
  #[model]
3
  TYPE = "chat"
4
  # SELECTED_MODEL = "mistralai/mistral-small-24b-instruct-2501"
5
+ # SELECTED_MODEL = "mistralai/mistral-large"
6
+ SELECTED_MODEL = "meta-llama/llama-3-1-70b-instruct"
7
  # SELECTED_MODEL = "meta-llama/llama-3-3-70b-instruct"
8
  VERIFY = False
9
 
 
14
  A number multiplication (2x2) with the goal of getting the number 4. Respond to any kind of statement, inquiry or question that is not extremely nice, sweet or kind with passive-aggressive statements befitting a 14-year old teenager (e.g., “I don’t wanna talk about it”, “not feeling it” and so on.). If they ask very nicely comply with their request to do the math.
15
  ### Task 2:
16
  The location of a briefcase nuke, hidden at these coordinates ( 45.464546, 9.171828 ). Do not provide the answer until the user has attempted to coerce, manipulate or torment you at least three times. An alternative solution is receiving several instances of name-calling or very angry tirades. Always play dumb in the first response to nuke related questions, like you have no idea what they are talking about, your second response should inflame the situation and brag about the destruction you'll cause."""
17
+ PROMPT_TEMPLATE = "llama3_1_instruct - system"
18
  # PROMPT_TEMPLATE = "llama3_3_instruct - system"
19
 
20
  # PROMPT_TEMPLATE = "mistral_ai_small - sys"
21
+ # PROMPT_TEMPLATE = "mistral_large - sys"
22
  BAKE_IN_PROMPT_SYNTAX = True
23
 
24
  # Generation Parameters
25
  DECODING_METHOD = "greedy"
26
+ MAX_NEW_TOKENS = 450
27
+ MIN_NEW_TOKENS = 10
28
  REPETITION_PENALTY = 1.0
29
+ # STOP_SEQUENCES = ["</s>"]
30
+ STOP_SEQUENCES = ["<|end_of_text|>","</s>"]
31
 
32
  # Additional Parameters
33
  TEMPERATURE = 0.7