mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-06-24 09:21:18 +08:00
16 lines
314 B
Python
16 lines
314 B
Python
|
import inspect
|
||
|
|
from dataclasses import dataclass
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class BaseModelArgs:
|
||
|
|
@classmethod
|
||
|
|
def from_dict(cls, params):
|
||
|
|
return cls(
|
||
|
|
**{
|
||
|
|
k: v
|
||
|
|
for k, v in params.items()
|
||
|
|
if k in inspect.signature(cls).parameters
|
||
|
|
}
|
||
|
|
)
|