Confusing forward Type Signature

[ncfrey]

Feedback: Forward signature is neither clearly static nor dynamic.

• Refine forward() method signature:
  • Decide between static: forward(self, input_ids: Tensor, attention_mask: Tensor)
  • Or dynamic: forward(self, inputs: Sequence[Tensor])
• Use typing annotations and docstrings to clarify expected inputs and outputs.