Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -27,7 +27,9 @@ pip install -e .
|
|
| 27 |
```python
|
| 28 |
from dmx.compressor.modeling import DmxModel
|
| 29 |
import lm_eval
|
|
|
|
| 30 |
|
|
|
|
| 31 |
model_args = "pretrained=d-matrix/bert-base,trust_remote_code=True"
|
| 32 |
|
| 33 |
lm = lm_eval.api.registry.get_model("hf").create_from_arg_string(model_args, {"batch_size": 1})
|
|
|
|
| 27 |
```python
|
| 28 |
from dmx.compressor.modeling import DmxModel
|
| 29 |
import lm_eval
|
| 30 |
+
from lm_eval.models.huggingface import HFLM
|
| 31 |
|
| 32 |
+
lm_eval.api.registry.register_model("hf", HFLM)
|
| 33 |
model_args = "pretrained=d-matrix/bert-base,trust_remote_code=True"
|
| 34 |
|
| 35 |
lm = lm_eval.api.registry.get_model("hf").create_from_arg_string(model_args, {"batch_size": 1})
|