bmah-dmx commited on
Commit
7e695e6
·
verified ·
1 Parent(s): bef6b65

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -48,12 +48,12 @@ pip install lm-eval
48
  from dmx.compressor.modeling import DmxModel
49
  import lm_eval
50
 
51
- model_args = f"pretrained="d-matrix/gpt2-xl",trust_remote_code=True"
52
 
53
  lm = lm_eval.api.registry.get_model("hf").create_from_arg_string(model_args, {"batch_size": 1})
54
 
55
  # Transform the model with DMX
56
  lm._model = DmxModel.from_torch(lm._model).to_basic_model() # Using BASIC configuration
57
 
58
- eval_results = lm_eval.evaluate(lm, lm_eval.tasks.get_task_dict([task]) # Assign desired task, i.e. "wikitext"
59
  ```
 
48
  from dmx.compressor.modeling import DmxModel
49
  import lm_eval
50
 
51
+ model_args = "pretrained='d-matrix/gpt2-xl',trust_remote_code=True"
52
 
53
  lm = lm_eval.api.registry.get_model("hf").create_from_arg_string(model_args, {"batch_size": 1})
54
 
55
  # Transform the model with DMX
56
  lm._model = DmxModel.from_torch(lm._model).to_basic_model() # Using BASIC configuration
57
 
58
+ eval_results = lm_eval.evaluate(lm, lm_eval.tasks.get_task_dict([task])) # Assign desired task, i.e. "wikitext"
59
  ```