Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
@@ -26,12 +26,12 @@ pip install lm-eval
|
|
26 |
from dmx.compressor.modeling import DmxModel
|
27 |
import lm_eval
|
28 |
|
29 |
-
model_args =
|
30 |
|
31 |
lm = lm_eval.api.registry.get_model("hf").create_from_arg_string(model_args, {"batch_size": 1})
|
32 |
|
33 |
# Transform the model with DMX
|
34 |
lm._model = DmxModel.from_torch(lm._model).to_basic_model() # Using BASIC configuration
|
35 |
|
36 |
-
eval_results = lm_eval.evaluate(lm, lm_eval.tasks.get_task_dict([task]) # Assign desired task, i.e. "wikitext"
|
37 |
```
|
|
|
26 |
from dmx.compressor.modeling import DmxModel
|
27 |
import lm_eval
|
28 |
|
29 |
+
model_args = "pretrained='d-matrix/bert-base',trust_remote_code=True"
|
30 |
|
31 |
lm = lm_eval.api.registry.get_model("hf").create_from_arg_string(model_args, {"batch_size": 1})
|
32 |
|
33 |
# Transform the model with DMX
|
34 |
lm._model = DmxModel.from_torch(lm._model).to_basic_model() # Using BASIC configuration
|
35 |
|
36 |
+
eval_results = lm_eval.evaluate(lm, lm_eval.tasks.get_task_dict([task])) # Assign desired task, i.e. "wikitext"
|
37 |
```
|