Spaces:
Runtime error
Runtime error
Dhairyashil Ghatage commited on
Commit ·
251429f
1
Parent(s): a9416f6
add base model
Browse files- models/base.py +15 -0
models/base.py
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import inspect
|
| 2 |
+
from dataclasses import dataclass
|
| 3 |
+
|
| 4 |
+
|
| 5 |
+
@dataclass
|
| 6 |
+
class BaseModelArgs:
|
| 7 |
+
@classmethod
|
| 8 |
+
def from_dict(cls, params):
|
| 9 |
+
return cls(
|
| 10 |
+
**{
|
| 11 |
+
k: v
|
| 12 |
+
for k, v in params.items()
|
| 13 |
+
if k in inspect.signature(cls).parameters
|
| 14 |
+
}
|
| 15 |
+
)
|