Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
@@ -27,6 +27,7 @@ pip install -e .
|
|
27 |
```python
|
28 |
from dmx.compressor.modeling import DmxModel
|
29 |
import lm_eval
|
|
|
30 |
|
31 |
lm_eval.api.registry.register_model("hf", HFLM)
|
32 |
model_args = "pretrained=d-matrix/bert-large,trust_remote_code=True"
|
|
|
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-large,trust_remote_code=True"
|