|
--- |
|
license: mit |
|
--- |
|
|
|
This is a personal reproduction of the Repoformer-1B model in [this paper](https://arxiv.org/pdf/2403.10059). This model is trained on ~300k Python chunk and function completion instances created from the Stack. |
|
|
|
You can follow the instructions [here](https://github.com/amazon-science/Repoformer) to use the model. |
|
|
|
If you find this useful, please cite: |
|
``` |
|
@inproceedings{DBLP:conf/icml/0054AZR024, |
|
author = {Di Wu and |
|
Wasi Uddin Ahmad and |
|
Dejiao Zhang and |
|
Murali Krishna Ramanathan and |
|
Xiaofei Ma}, |
|
title = {Repoformer: Selective Retrieval for Repository-Level Code Completion}, |
|
booktitle = {Forty-first International Conference on Machine Learning, {ICML} 2024, |
|
Vienna, Austria, July 21-27, 2024}, |
|
publisher = {OpenReview.net}, |
|
year = {2024}, |
|
url = {https://openreview.net/forum?id=moyG54Okrj}, |
|
timestamp = {Mon, 02 Sep 2024 16:55:25 +0200}, |
|
biburl = {https://dblp.org/rec/conf/icml/0054AZR024.bib}, |
|
bibsource = {dblp computer science bibliography, https://dblp.org} |
|
} |
|
``` |