Update configuration_quiet.py

configuration_quiet.py

@@ -114,6 +114,7 @@ class QuietConfig(PretrainedConfig):
         max_thoughts: int = 3,
         thought_length: int = 10,
         eos_token_id=2,
+        end_token_id=2,
         start_token_id=0,  # Add this line
         tie_word_embeddings=False,
         rope_theta=10000.0,
@@ -147,7 +148,8 @@ class QuietConfig(PretrainedConfig):
             pad_token_id=pad_token_id,
             bos_token_id=bos_token_id,
             eos_token_id=eos_token_id,
-            start_token_id=start_token_id,  # Add this line
+            start_token_id=start_token_id,
+            end_token_id=end_token_id,# Add this line
             tie_word_embeddings=tie_word_embeddings,
             **kwargs,
         )