Update README.md
Browse files
README.md
CHANGED
@@ -5,7 +5,7 @@ tags: []
|
|
5 |
|
6 |
# Model Card for mAInframer-1
|
7 |
|
8 |
-
|
9 |
|
10 |
## Model Details
|
11 |
|
@@ -13,15 +13,12 @@ Bloob model for auto-complete in the Cobol programming language.
|
|
13 |
|
14 |
mAInframer is a series of models with different parameter counts: 7b, 13b, 34b.
|
15 |
|
16 |
-
These models are pretrained on code (
|
17 |
|
18 |
|
19 |
-
- **Developed by: Bloop**
|
20 |
-
- **Finetuned from model: Codellama**
|
21 |
-
|
22 |
## Uses
|
23 |
|
24 |
-
This model can be used for code completion and fill-in-the middle (
|
25 |
|
26 |
## How to Get Started with the Model
|
27 |
|
@@ -121,7 +118,7 @@ Result:
|
|
121 |
|
122 |
Follow the format: `<PRE>prefix<SUF>suffix<MID>`
|
123 |
|
124 |
-
To complete `PROCEDURE DIVISION` and infill `WORKING STORAGE SECTION.` to solve [
|
125 |
|
126 |
```python
|
127 |
prompt = '''<PRE> IDENTIFICATION DIVISION.
|
@@ -197,16 +194,12 @@ Result:
|
|
197 |
|
198 |
## Training Details
|
199 |
|
200 |
-
Base model:
|
201 |
Finetuning type: LoRA
|
202 |
|
203 |
-
###
|
204 |
-
|
205 |
-
Stack-v2(Cobol) + Stack Overflow (Cobol) + Web scraped Cobol + Synthetic Cobol data
|
206 |
-
|
207 |
-
#### Metrics
|
208 |
|
209 |
-
[
|
210 |
|
211 |
| **Model** | CobolEval (pass@1) |
|
212 |
|----------------------|--------------------|
|
|
|
5 |
|
6 |
# Model Card for mAInframer-1
|
7 |
|
8 |
+
LLM for the COBOL programming language.
|
9 |
|
10 |
## Model Details
|
11 |
|
|
|
13 |
|
14 |
mAInframer is a series of models with different parameter counts: 7b, 13b, 34b.
|
15 |
|
16 |
+
These models are pretrained on code (CodeLlama base models) and fine-tuned on COBOL code.
|
17 |
|
18 |
|
|
|
|
|
|
|
19 |
## Uses
|
20 |
|
21 |
+
This model can be used for code completion and fill-in-the middle (COBOL).
|
22 |
|
23 |
## How to Get Started with the Model
|
24 |
|
|
|
118 |
|
119 |
Follow the format: `<PRE>prefix<SUF>suffix<MID>`
|
120 |
|
121 |
+
To complete `PROCEDURE DIVISION` and infill `WORKING STORAGE SECTION.` to solve [COBOLEval](https://github.com/BloopAI/COBOLEval) problems:
|
122 |
|
123 |
```python
|
124 |
prompt = '''<PRE> IDENTIFICATION DIVISION.
|
|
|
194 |
|
195 |
## Training Details
|
196 |
|
197 |
+
Base model: CodeLlama
|
198 |
Finetuning type: LoRA
|
199 |
|
200 |
+
### Metrics
|
|
|
|
|
|
|
|
|
201 |
|
202 |
+
[COBOLEval](https://github.com/BloopAI/COBOLEval) is an adaptation of HumanEval where the problems are translated to COBOL.
|
203 |
|
204 |
| **Model** | CobolEval (pass@1) |
|
205 |
|----------------------|--------------------|
|