phi_2_QLoRA / models /base.py
Dhairyashil Ghatage
add base model
251429f
raw
history blame contribute delete
314 Bytes
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
}
)