Skip to content

Commit

Permalink
Update README.md with the citation section
Browse files Browse the repository at this point in the history
  • Loading branch information
hepengfe authored Jan 15, 2025
1 parent ab29bde commit 51caa85
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -42,3 +42,18 @@ This codebase is highly optimized for hfai platform, and it supports the followi
Here are some extra notes for hfai platform:
* default ni dataset dir is `../../data` due to hfai compatibility.
* **Pytorch and CUDA compatibility**: hfai platform has CUDA version 11.3, and [peft](https://github.com/huggingface/peft) setup requires `torch>=1.13.0`. Therefore, we use the corresponding pytorch version `1.10.2+cu113` by peft setup.


## Citation
If you found the codebase or my work valuable, please cite:
```
@misc{he2024parameterefficientinstructiontuning,
title={Parameter Efficient Instruction Tuning: An Empirical Study},
author={Pengfei He},
year={2024},
eprint={2411.16775},
archivePrefix={arXiv},
primaryClass={cs.CL},
url={https://arxiv.org/abs/2411.16775},
}
```

0 comments on commit 51caa85

Please sign in to comment.