Text Generation
Transformers
English
codegen
Inference Endpoints
codegen-6B-lora / README.md
mhhmm's picture
Update README.md
e9cdda5
---
license: mit
datasets:
- mhhmm/leetcode-solutions-python
- deepmind/code_contests
language:
- en
library_name: transformers
pipeline_tag: text-generation
widget:
- text: "
# Given an array of integers, return indices of the two numbers such that they add up to a specific target
def twoSum(array, target) -> List[int]:
"
example_title: "Twosum problem"
---
LLM: [Salesforce/CodeGen-6B-Mono](https://huggingface.co/Salesforce/codegen-6B-mono)
I'm using [Peft](https://github.com/huggingface/peft) for tuning
Tuning:
- [LoRA](https://github.com/microsoft/LoRA)
- [Leetcode](https://huggingface.co/datasets/mhhmm/leetcode-solutions-python)
- [Google Deepmind Code contests](https://huggingface.co/datasets/deepmind/code_contests)
- Google Colab Pro+ in ~2 hours, shoutout to my friend TieuPhuong