Methodforward(self, x, position_embeddings, past_key_value=None, use_cache=False, attention_mask=None)
model/model_minimind.py:111
Methodforward(self, hidden_states, position_embeddings, past_key_value=None, use_cache=False, attention_mask=None)
model/model_minimind.py:186
Methodforward(self, input_ids, attention_mask=None, past_key_values=None, use_cache=False, **kwargs)
model/model_minimind.py:209
Methodrollout(self, prompt_ids: Tensor, attention_mask: Tensor, num_generations: int, max_new_tokens: int, temperature: flo
trainer/rollout_engine.py:71
Methodrollout(self, prompt_ids: Tensor, attention_mask: Tensor, num_generations: int, max_new_tokens: int, temperature: flo
trainer/rollout_engine.py:107