Update modeling_quiet.py

modeling_quiet.py

@@ -1215,7 +1215,7 @@ class QuietForCausalLM(QuietPreTrainedModel):
         logits = self.lm_head(hidden_states)
 
 
-        thought_ids, thought_embeddings = self._generate_thoughts(hidden_states, max_length=self.config.max_thought_length)
+        thought_ids, thought_embeddings = self._generate_thoughts(hidden_states, max_length=self.config.thought_length)
         thought_hidden_states = self.model(inputs_embeds=thought_embeddings).last_hidden_state
 
         # Compute thought logits