Commit
·
a03a283
0
Parent(s):
initial commit
Browse files- .gitattributes +1 -0
- added_tokens.json +4 -0
- config.json +31 -0
- merges.txt +0 -0
- model.safetensors +3 -0
- readme.md +50 -0
- special_tokens_map.json +58 -0
- tokenizer.json +0 -0
- tokenizer_config.json +204 -0
- vocab.json +0 -0
.gitattributes
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
model.safetensors filter=lfs diff=lfs merge=lfs -text
|
added_tokens.json
ADDED
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"<mask>": 49152,
|
3 |
+
"<pad>": 49153
|
4 |
+
}
|
config.json
ADDED
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "Partha117/blaze",
|
3 |
+
"activation_function": "gelu_new",
|
4 |
+
"architectures": [
|
5 |
+
"CodeSageModel"
|
6 |
+
],
|
7 |
+
"attention_dropout_prob": 0.1,
|
8 |
+
"auto_map": {
|
9 |
+
"AutoConfig": "codesage/codesage-small--config_codesage.CodeSageConfig",
|
10 |
+
"AutoModel": "codesage/codesage-small--modeling_codesage.CodeSageModel",
|
11 |
+
"AutoModelForMaskedLM": "codesage/codesage-small--modeling_codesage.CodeSageForMaskedLM",
|
12 |
+
"AutoModelForSequenceClassification": "codesage/codesage-small--modeling_codesage.CodeSageForSequenceClassification"
|
13 |
+
},
|
14 |
+
"bos_token_id": 0,
|
15 |
+
"embedding_dropout_prob": 0.1,
|
16 |
+
"eos_token_id": 0,
|
17 |
+
"hidden_size": 1024,
|
18 |
+
"initializer_range": 0.02,
|
19 |
+
"intermediate_size": 4096,
|
20 |
+
"layer_norm_epsilon": 1e-05,
|
21 |
+
"max_position_embeddings": 2048,
|
22 |
+
"model_type": "codesage",
|
23 |
+
"num_attention_heads": 8,
|
24 |
+
"num_hidden_layers": 6,
|
25 |
+
"pad_token_id": 49153,
|
26 |
+
"position_embedding_type": "absolute",
|
27 |
+
"residual_dropout_prob": 0.1,
|
28 |
+
"torch_dtype": "float32",
|
29 |
+
"transformers_version": "4.49.0",
|
30 |
+
"vocab_size": 49154
|
31 |
+
}
|
merges.txt
ADDED
The diff for this file is too large to render.
See raw diff
|
|
model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:15363f2a8398bceeecb427041e20560abfae5dda0ebef83a52162a417575b920
|
3 |
+
size 512047808
|
readme.md
ADDED
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Blaze (Finetuned Variant)
|
2 |
+
|
3 |
+
This model is a finetuned version of the **BLAZE** model described in the paper:
|
4 |
+
|
5 |
+
**_BLAZE: Cross‑Language and Cross‑Project Bug Localization via Dynamic Chunking and Hard Example Learning_**
|
6 |
+
DOI: [10.5281/zenodo.15122980](https://doi.org/10.5281/zenodo.15122980) :contentReference[oaicite:1]{index=1}
|
7 |
+
|
8 |
+
---
|
9 |
+
|
10 |
+
## 📘 What’s Inside
|
11 |
+
|
12 |
+
A Transformer-based bug localization model fine-tuned on additional cross-project datasets to enhance its ability to pinpoint bugs in unseen codebases.
|
13 |
+
|
14 |
+
---
|
15 |
+
|
16 |
+
## 🧪 Fine-tuning Details
|
17 |
+
|
18 |
+
- **Starting point:** Pretrained Blaze model from Zenodo.
|
19 |
+
- **Enhancements:** Further trained on a curated dataset using dynamic chunking and hard-negative sampling — all detailed in the original manuscript and accompanying code release.
|
20 |
+
|
21 |
+
---
|
22 |
+
|
23 |
+
## 🔍 Intended Usage
|
24 |
+
|
25 |
+
- **Primary task:** Automatic identification of buggy code segments across languages and projects.
|
26 |
+
- **How to use:** Load the model and feed it code snippets to receive localized bug predictions.
|
27 |
+
|
28 |
+
---
|
29 |
+
|
30 |
+
## 📄 Citation
|
31 |
+
|
32 |
+
If you use this model, please cite:
|
33 |
+
|
34 |
+
> _BLAZE: Cross‑Language and Cross‑Project Bug Localization via Dynamic Chunking and Hard Example Learning_, available at DOI: 10.5281/zenodo.15122980 :contentReference[oaicite:2]{index=2}
|
35 |
+
|
36 |
+
---
|
37 |
+
|
38 |
+
## 📁 Contents of This Repo
|
39 |
+
|
40 |
+
- `config.json`
|
41 |
+
- `pytorch_model.bin` (or `tf_model.h5`)
|
42 |
+
- `tokenizer.json` (if applicable)
|
43 |
+
- `README.md`
|
44 |
+
|
45 |
+
---
|
46 |
+
|
47 |
+
## 🛠️ Setup
|
48 |
+
|
49 |
+
```bash
|
50 |
+
pip install transformers huggingface_hub
|
special_tokens_map.json
ADDED
@@ -0,0 +1,58 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"additional_special_tokens": [
|
3 |
+
"<|endoftext|>",
|
4 |
+
"<fim_prefix>",
|
5 |
+
"<fim_middle>",
|
6 |
+
"<fim_suffix>",
|
7 |
+
"<fim_pad>",
|
8 |
+
"<filename>",
|
9 |
+
"<gh_stars>",
|
10 |
+
"<issue_start>",
|
11 |
+
"<issue_comment>",
|
12 |
+
"<issue_closed>",
|
13 |
+
"<jupyter_start>",
|
14 |
+
"<jupyter_text>",
|
15 |
+
"<jupyter_code>",
|
16 |
+
"<jupyter_output>",
|
17 |
+
"<empty_output>",
|
18 |
+
"<commit_before>",
|
19 |
+
"<commit_msg>",
|
20 |
+
"<commit_after>",
|
21 |
+
"<reponame>"
|
22 |
+
],
|
23 |
+
"bos_token": {
|
24 |
+
"content": "<|endoftext|>",
|
25 |
+
"lstrip": false,
|
26 |
+
"normalized": false,
|
27 |
+
"rstrip": false,
|
28 |
+
"single_word": false
|
29 |
+
},
|
30 |
+
"eos_token": {
|
31 |
+
"content": "<|endoftext|>",
|
32 |
+
"lstrip": false,
|
33 |
+
"normalized": false,
|
34 |
+
"rstrip": false,
|
35 |
+
"single_word": false
|
36 |
+
},
|
37 |
+
"mask_token": {
|
38 |
+
"content": "<mask>",
|
39 |
+
"lstrip": false,
|
40 |
+
"normalized": false,
|
41 |
+
"rstrip": false,
|
42 |
+
"single_word": false
|
43 |
+
},
|
44 |
+
"pad_token": {
|
45 |
+
"content": "<pad>",
|
46 |
+
"lstrip": false,
|
47 |
+
"normalized": false,
|
48 |
+
"rstrip": false,
|
49 |
+
"single_word": false
|
50 |
+
},
|
51 |
+
"unk_token": {
|
52 |
+
"content": "<|endoftext|>",
|
53 |
+
"lstrip": false,
|
54 |
+
"normalized": false,
|
55 |
+
"rstrip": false,
|
56 |
+
"single_word": false
|
57 |
+
}
|
58 |
+
}
|
tokenizer.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
tokenizer_config.json
ADDED
@@ -0,0 +1,204 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"add_prefix_space": false,
|
3 |
+
"added_tokens_decoder": {
|
4 |
+
"0": {
|
5 |
+
"content": "<|endoftext|>",
|
6 |
+
"lstrip": false,
|
7 |
+
"normalized": false,
|
8 |
+
"rstrip": false,
|
9 |
+
"single_word": false,
|
10 |
+
"special": true
|
11 |
+
},
|
12 |
+
"1": {
|
13 |
+
"content": "<fim_prefix>",
|
14 |
+
"lstrip": false,
|
15 |
+
"normalized": false,
|
16 |
+
"rstrip": false,
|
17 |
+
"single_word": false,
|
18 |
+
"special": true
|
19 |
+
},
|
20 |
+
"2": {
|
21 |
+
"content": "<fim_middle>",
|
22 |
+
"lstrip": false,
|
23 |
+
"normalized": false,
|
24 |
+
"rstrip": false,
|
25 |
+
"single_word": false,
|
26 |
+
"special": true
|
27 |
+
},
|
28 |
+
"3": {
|
29 |
+
"content": "<fim_suffix>",
|
30 |
+
"lstrip": false,
|
31 |
+
"normalized": false,
|
32 |
+
"rstrip": false,
|
33 |
+
"single_word": false,
|
34 |
+
"special": true
|
35 |
+
},
|
36 |
+
"4": {
|
37 |
+
"content": "<fim_pad>",
|
38 |
+
"lstrip": false,
|
39 |
+
"normalized": false,
|
40 |
+
"rstrip": false,
|
41 |
+
"single_word": false,
|
42 |
+
"special": true
|
43 |
+
},
|
44 |
+
"5": {
|
45 |
+
"content": "<filename>",
|
46 |
+
"lstrip": false,
|
47 |
+
"normalized": false,
|
48 |
+
"rstrip": false,
|
49 |
+
"single_word": false,
|
50 |
+
"special": true
|
51 |
+
},
|
52 |
+
"6": {
|
53 |
+
"content": "<gh_stars>",
|
54 |
+
"lstrip": false,
|
55 |
+
"normalized": false,
|
56 |
+
"rstrip": false,
|
57 |
+
"single_word": false,
|
58 |
+
"special": true
|
59 |
+
},
|
60 |
+
"7": {
|
61 |
+
"content": "<issue_start>",
|
62 |
+
"lstrip": false,
|
63 |
+
"normalized": false,
|
64 |
+
"rstrip": false,
|
65 |
+
"single_word": false,
|
66 |
+
"special": true
|
67 |
+
},
|
68 |
+
"8": {
|
69 |
+
"content": "<issue_comment>",
|
70 |
+
"lstrip": false,
|
71 |
+
"normalized": false,
|
72 |
+
"rstrip": false,
|
73 |
+
"single_word": false,
|
74 |
+
"special": true
|
75 |
+
},
|
76 |
+
"9": {
|
77 |
+
"content": "<issue_closed>",
|
78 |
+
"lstrip": false,
|
79 |
+
"normalized": false,
|
80 |
+
"rstrip": false,
|
81 |
+
"single_word": false,
|
82 |
+
"special": true
|
83 |
+
},
|
84 |
+
"10": {
|
85 |
+
"content": "<jupyter_start>",
|
86 |
+
"lstrip": false,
|
87 |
+
"normalized": false,
|
88 |
+
"rstrip": false,
|
89 |
+
"single_word": false,
|
90 |
+
"special": true
|
91 |
+
},
|
92 |
+
"11": {
|
93 |
+
"content": "<jupyter_text>",
|
94 |
+
"lstrip": false,
|
95 |
+
"normalized": false,
|
96 |
+
"rstrip": false,
|
97 |
+
"single_word": false,
|
98 |
+
"special": true
|
99 |
+
},
|
100 |
+
"12": {
|
101 |
+
"content": "<jupyter_code>",
|
102 |
+
"lstrip": false,
|
103 |
+
"normalized": false,
|
104 |
+
"rstrip": false,
|
105 |
+
"single_word": false,
|
106 |
+
"special": true
|
107 |
+
},
|
108 |
+
"13": {
|
109 |
+
"content": "<jupyter_output>",
|
110 |
+
"lstrip": false,
|
111 |
+
"normalized": false,
|
112 |
+
"rstrip": false,
|
113 |
+
"single_word": false,
|
114 |
+
"special": true
|
115 |
+
},
|
116 |
+
"14": {
|
117 |
+
"content": "<empty_output>",
|
118 |
+
"lstrip": false,
|
119 |
+
"normalized": false,
|
120 |
+
"rstrip": false,
|
121 |
+
"single_word": false,
|
122 |
+
"special": true
|
123 |
+
},
|
124 |
+
"15": {
|
125 |
+
"content": "<commit_before>",
|
126 |
+
"lstrip": false,
|
127 |
+
"normalized": false,
|
128 |
+
"rstrip": false,
|
129 |
+
"single_word": false,
|
130 |
+
"special": true
|
131 |
+
},
|
132 |
+
"16": {
|
133 |
+
"content": "<commit_msg>",
|
134 |
+
"lstrip": false,
|
135 |
+
"normalized": false,
|
136 |
+
"rstrip": false,
|
137 |
+
"single_word": false,
|
138 |
+
"special": true
|
139 |
+
},
|
140 |
+
"17": {
|
141 |
+
"content": "<commit_after>",
|
142 |
+
"lstrip": false,
|
143 |
+
"normalized": false,
|
144 |
+
"rstrip": false,
|
145 |
+
"single_word": false,
|
146 |
+
"special": true
|
147 |
+
},
|
148 |
+
"18": {
|
149 |
+
"content": "<reponame>",
|
150 |
+
"lstrip": false,
|
151 |
+
"normalized": false,
|
152 |
+
"rstrip": false,
|
153 |
+
"single_word": false,
|
154 |
+
"special": true
|
155 |
+
},
|
156 |
+
"49152": {
|
157 |
+
"content": "<mask>",
|
158 |
+
"lstrip": false,
|
159 |
+
"normalized": false,
|
160 |
+
"rstrip": false,
|
161 |
+
"single_word": false,
|
162 |
+
"special": true
|
163 |
+
},
|
164 |
+
"49153": {
|
165 |
+
"content": "<pad>",
|
166 |
+
"lstrip": false,
|
167 |
+
"normalized": false,
|
168 |
+
"rstrip": false,
|
169 |
+
"single_word": false,
|
170 |
+
"special": true
|
171 |
+
}
|
172 |
+
},
|
173 |
+
"additional_special_tokens": [
|
174 |
+
"<|endoftext|>",
|
175 |
+
"<fim_prefix>",
|
176 |
+
"<fim_middle>",
|
177 |
+
"<fim_suffix>",
|
178 |
+
"<fim_pad>",
|
179 |
+
"<filename>",
|
180 |
+
"<gh_stars>",
|
181 |
+
"<issue_start>",
|
182 |
+
"<issue_comment>",
|
183 |
+
"<issue_closed>",
|
184 |
+
"<jupyter_start>",
|
185 |
+
"<jupyter_text>",
|
186 |
+
"<jupyter_code>",
|
187 |
+
"<jupyter_output>",
|
188 |
+
"<empty_output>",
|
189 |
+
"<commit_before>",
|
190 |
+
"<commit_msg>",
|
191 |
+
"<commit_after>",
|
192 |
+
"<reponame>"
|
193 |
+
],
|
194 |
+
"bos_token": "<|endoftext|>",
|
195 |
+
"clean_up_tokenization_spaces": true,
|
196 |
+
"eos_token": "<|endoftext|>",
|
197 |
+
"extra_special_tokens": {},
|
198 |
+
"mask_token": "<mask>",
|
199 |
+
"model_max_length": 1000000000000000019884624838656,
|
200 |
+
"pad_token": "<pad>",
|
201 |
+
"tokenizer_class": "GPT2Tokenizer",
|
202 |
+
"unk_token": "<|endoftext|>",
|
203 |
+
"vocab_size": 49152
|
204 |
+
}
|
vocab.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|