Update config.py

config.py

@@ -54,6 +54,16 @@ INFERENCE_REPETITION_PENALTY = 1.2
 INFERENCE_TEMPERATURE = 0.7
 INFERENCE_TOP_K = 50
 
+# Generation/System Prompt
+SYSTEM_MSG = "You are a motion generation assistant. Generate motion tokens for the given text description."
+GEN_MAX_NEW_TOKENS = 100
+GEN_TEMPERATURE = 0.7
+GEN_TOP_P = 0.9
+GEN_TOP_K = 50
+GEN_NO_REPEAT_NGRAM_SIZE = 3
+GEN_REPETITION_PENALTY = 1.2
+GEN_END_LOGIT_SLOPE = 0.1
+
 # Special Tokens (matches test_overfit.py)
 M_START = "<M_START>"
 M_END = "<M_END>"