bmah-dmx commited on
Commit
c643dab
·
verified ·
1 Parent(s): 873c930

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +1 -0
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"