jinjieyuan's picture
Update README.md
42435b8 verified
metadata
language: en
license: apache-2.0
library_name: transformers

SQFT Fine-tuned Model: sqft-sparsepeft-phi-3-mini-4k-50-cs-heu

Evaluation

MODEL_NAME=IntelLabs/sqft-sparsepeft-phi-3-mini-4k-50-cs-heu
lm_eval --model hf --model_args pretrained=${MODEL_NAME},add_bos_token=True,trust_remote_code=True --tasks piqa,arc_easy,arc_challenge,hellaswag,openbookqa,boolq,winogrande --batch_size auto:4

Refer to our repo for the environment information to run this command.

Model Sources

Citation

@article{munoz2024sqft,
  title = {SQFT: Low-cost Model Adaptation in Low-precision Sparse Foundation Models},
  author={J. Pablo Munoz and Jinjie Yuan and Nilesh Jain},
  journal={The 2024 Conference on Empirical Methods in Natural Language Processing (Findings)},
  year={2024}
}

License

Apache-2.0