Fix infinite recursion in LoRA module repr when inside interceptor

[arJ-V]

Problem

The LoRA wrapper classes (LoRADense, LoRAEinsum, LoRADenseGeneral) caused infinite recursion when repr() was called inside a ModuleInterceptor context. This occurred because the default __repr__ implementation called repr(self.wrapped), which could re-trigger the interceptor and create a recursion loop.

Previous State

• Calling repr() on LoRA modules inside interceptors caused a stack overflow
• Made debugging and inspection of LoRA-wrapped modules difficult
• A TODO comment indicated this was a known issue

Solution

Added custom __repr__ methods to all three LoRA wrapper classes that:

• Use type(self.wrapped).__name__ instead of repr(self.wrapped) to avoid recursion
• Display useful information (rank, wrapped module type, and key attributes)
• Safely access simple attributes without triggering interceptors

Implementation

• LoRADense.__repr__()

  • Shows rank and wrapped Dense features

• LoRAEinsum.__repr__()

  • Shows rank and wrapped Einsum einsum_str / shape

• LoRADenseGeneral.__repr__()

  • Shows rank and wrapped DenseGeneral features / axis / batch_dims

PR opened for changes: #474

[pannaga-33]

Hi @arJ-V Thanks for submitting your Pull Request to Gemma!
We value your contribution and appreciate your patience while it is reviewed.