mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-08-11 19:56:42 +08:00
Update llms/mlx_lm/models/gemma3_text.py
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
This commit is contained in:
parent
f034a7b72a
commit
92b06ddcad
@ -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):
|
||||||
|
Loading…
Reference in New Issue
Block a user