Update llms/mlx_lm/models/gemma3_text.py

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
This commit is contained in:
Prince Canuma 2025-03-12 15:00:26 +01:00 committed by GitHub
parent f034a7b72a
commit 92b06ddcad
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -27,15 +27,6 @@ class ModelArgs(BaseModelArgs):
sliding_window: int = 512 sliding_window: int = 512
sliding_window_pattern: int = 6 sliding_window_pattern: int = 6
@classmethod
def from_dict(cls, params):
return cls(
**{
k: v
for k, v in params.items()
if k in inspect.signature(cls).parameters
}
)
class Attention(nn.Module): class Attention(nn.Module):