Upload README.md with huggingface_hub

README.md

@@ -28,12 +28,12 @@ pip install -e .
 from dmx.compressor.modeling import DmxModel
 import lm_eval
 
-model_args = "pretrained='d-matrix/bert-base',trust_remote_code=True"
+model_args = "pretrained=d-matrix/bert-base,trust_remote_code=True"
 
 lm = lm_eval.api.registry.get_model("hf").create_from_arg_string(model_args, {"batch_size": 1})
 
 # Transform the model with DMX
-lm._model = DmxModel.from_torch(lm._model).to_basic_model()  # Using BASIC configuration
+lm._model = DmxModel.from_torch(lm._model)
 
-eval_results = lm_eval.evaluate(lm, lm_eval.tasks.get_task_dict([task]))  # Assign desired task, i.e. "wikitext"
+eval_results = lm_eval.evaluate(lm, lm_eval.tasks.get_task_dict(["wikitext"]))  # Assign desired task, i.e. "wikitext"
 ```