Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
# CodeLlama-70B_for_NTR
|
2 |
|
3 |
We fine-tuned [CodeLlama-70B](https://huggingface.co/codellama/CodeLlama-70b-hf) on [Transfer_dataset](https://drive.google.com/drive/folders/1F1BPfTxHDGX-OCBthudCbu_6Qvcg_fbP?usp=drive_link) under the [NTR](https://sites.google.com/view/neuraltemplaterepair) framework for APR research.
|
@@ -108,11 +115,11 @@ generated_ids = model.generate(
|
|
108 |
for generated_id in generated_ids:
|
109 |
generated_text = tokenizer.decode(generated_id, skip_special_tokens=False)
|
110 |
patch = generated_text.split(E_INST)[1]
|
111 |
-
patch =
|
112 |
print(patch)
|
113 |
|
114 |
|
115 |
```
|
116 |
|
117 |
## Model Details
|
118 |
-
*Note: Use of this model is governed by the Meta license. Meta developed and publicly released the Code Llama family of large language models (LLMs).
|
|
|
1 |
+
---
|
2 |
+
license: llama2
|
3 |
+
pipeline_tag: text-generation
|
4 |
+
tags:
|
5 |
+
- code
|
6 |
+
- automated program repair
|
7 |
+
---
|
8 |
# CodeLlama-70B_for_NTR
|
9 |
|
10 |
We fine-tuned [CodeLlama-70B](https://huggingface.co/codellama/CodeLlama-70b-hf) on [Transfer_dataset](https://drive.google.com/drive/folders/1F1BPfTxHDGX-OCBthudCbu_6Qvcg_fbP?usp=drive_link) under the [NTR](https://sites.google.com/view/neuraltemplaterepair) framework for APR research.
|
|
|
115 |
for generated_id in generated_ids:
|
116 |
generated_text = tokenizer.decode(generated_id, skip_special_tokens=False)
|
117 |
patch = generated_text.split(E_INST)[1]
|
118 |
+
patch = patch.replace(tokenizer.eos_token,'')
|
119 |
print(patch)
|
120 |
|
121 |
|
122 |
```
|
123 |
|
124 |
## Model Details
|
125 |
+
*Note: Use of this model is governed by the Meta license. Meta developed and publicly released the Code Llama family of large language models (LLMs).
|