aaabiao commited on
Commit
5e0f2dd
1 Parent(s): c17376a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +98 -0
README.md CHANGED
@@ -1,3 +1,101 @@
1
  ---
 
 
 
 
 
2
  license: apache-2.0
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ language:
3
+ - en
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - code
7
  license: apache-2.0
8
  ---
9
+
10
+ <h1 align="center"> OpenCodeInterpreter: Integrating Code Generation with Execution and Refinement<h1>
11
+
12
+ <p align="center">
13
+ <img width="1000px" alt="OpenCodeInterpreter" src="https://opencodeinterpreter.github.io/static/images/figure1.png">
14
+ </p>
15
+ <p align="center">
16
+ <a href="https://opencodeinterpreter.github.io/">[🏠Homepage]</a>
17
+ |
18
+ <a href="https://github.com/OpenCodeInterpreter/OpenCodeInterpreter/">[🛠️Code]</a>
19
+ </p>
20
+ <hr>
21
+
22
+ ## Introduction
23
+ OpenCodeInterpreter is a family of open-source code generation systems designed to bridge the gap between large language models and advanced proprietary systems like the GPT-4 Code Interpreter. It significantly advances code generation capabilities by integrating execution and iterative refinement functionalities.
24
+
25
+ For further information and related work, refer to our paper: ["OpenCodeInterpreter: A System for Enhanced Code Generation and Execution"](https://arxiv.org/abs/2402.14658) available on arXiv.
26
+
27
+ ## Model Information
28
+ This model is based on [starcoder2-15b](https://huggingface.co/bigcode/starcoder2-15b).
29
+
30
+ ## Benchmark Scores
31
+
32
+ The OpenCodeInterpreter Models series exemplifies the evolution of coding model performance, particularly highlighting the significant enhancements brought about by the integration of execution feedback. In an effort to quantify these improvements, we present a detailed comparison across two critical benchmarks: HumanEval and MBPP. This comparison not only showcases the individual performance metrics on each benchmark but also provides an aggregated view of the overall performance enhancement. The subsequent table succinctly encapsulates the performance data, offering a clear perspective on how execution feedback contributes to elevating the models' capabilities in code interpretation and execution tasks.
33
+
34
+ | **Benchmark** | **HumanEval (+)** | **MBPP (+)** | **Average (+)** |
35
+ |---------------|-------------------|--------------|-----------------|
36
+ | **OpenCodeInterpreter-DS-1.3B** | 65.2 (61.0) | 63.4 (52.4) | 64.3 (56.7) |
37
+ | + Execution Feedback | 65.2 (62.2) | 65.2 (55.6) | 65.2 (58.9) |
38
+ | **OpenCodeInterpreter-DS-6.7B** | 76.2 (72.0) | 73.9 (63.7) | 75.1 (67.9) |
39
+ | + Execution Feedback | 81.1 (78.7) | 82.7 (72.4) | 81.9 (75.6) |
40
+ | + Synth. Human Feedback | 87.2 (86.6) | 86.2 (74.2) | 86.7 (80.4) |
41
+ | + Synth. Human Feedback (Oracle) | 89.7 (86.6) | 87.2 (75.2) | 88.5 (80.9) |
42
+ | **OpenCodeInterpreter-DS-33B** | 79.3 (74.3) | 78.7 (66.4) | 79.0 (70.4) |
43
+ | + Execution Feedback | 82.9 (80.5) | 83.5 (72.2) | 83.2 (76.4) |
44
+ | + Synth. Human Feedback | 88.4 (86.0) | 87.5 (75.9) | 88.0 (81.0) |
45
+ | + Synth. Human Feedback (Oracle) | 92.7 (89.7) | 90.5 (79.5) | 91.6 (84.6) |
46
+ | **OpenCodeInterpreter-CL-7B** | 72.6 (67.7) | 66.4 (55.4) | 69.5 (61.6) |
47
+ | + Execution Feedback | 75.6 (70.1) | 69.9 (60.7) | 72.8 (65.4) |
48
+ | **OpenCodeInterpreter-CL-13B** | 77.4 (73.8) | 70.7 (59.2) | 74.1 (66.5) |
49
+ | + Execution Feedback | 81.1 (76.8) | 78.2 (67.2) | 79.7 (72.0) |
50
+ | **OpenCodeInterpreter-CL-34B** | 78.0 (72.6) | 73.4 (61.4) | 75.7 (67.0) |
51
+ | + Execution Feedback | 81.7 (78.7) | 80.2 (67.9) | 81.0 (73.3) |
52
+ | **OpenCodeInterpreter-CL-70B** | 76.2 (70.7) | 73.0 (61.9) | 74.6 (66.3) |
53
+ | + Execution Feedback | 79.9 (77.4) | 81.5 (69.9) | 80.7 (73.7) |
54
+ | **OpenCodeInterpreter-GM-7B** | 56.1 (50.0) | 39.8 (34.6) | 48.0 (42.3) |
55
+ | + Execution Feedback | 64.0 (54.3) | 48.6 (40.9) | 56.3 (47.6) |
56
+ | **OpenCodeInterpreter-SC2-3B** | 65.2 (57.9) | 62.7 (52.9) | 64.0 (55.4) |
57
+ | + Execution Feedback | 67.1 (60.4) | 63.4 (54.9) | 65.3 (57.7) |
58
+ | **OpenCodeInterpreter-SC2-7B** | 73.8 (68.9) | 61.7 (51.1) | 67.8 (60.0) |
59
+ | + Execution Feedback | 75.6 (69.5) | 66.9 (55.4) | 71.3 (62.5) |
60
+ | **OpenCodeInterpreter-SC2-15B** | 75.6 (69.5) | 71.2 (61.2) | 73.4 (65.4) |
61
+ | + Execution Feedback | 77.4 (72.0) | 74.2 (63.4) | 75.8 (67.7) |
62
+
63
+ *Note: The "(+)" notation represents scores from extended versions of the HumanEval and MBPP benchmarks. To ensure a fair comparison, the results shown for adding execution feedback are based on outcomes after just one iteration of feedback, without unrestricted iterations. This approach highlights the immediate impact of execution feedback on performance improvements across benchmarks.*
64
+
65
+
66
+ ## Model Usage
67
+ ### Inference
68
+
69
+ ```python
70
+ import torch
71
+ from transformers import AutoTokenizer, AutoModelForCausalLM
72
+ model_path="m-a-p/OpenCodeInterpreter-SC2-3B"
73
+
74
+ tokenizer = AutoTokenizer.from_pretrained(model_path)
75
+ model = AutoModelForCausalLM.from_pretrained(
76
+ model_path,
77
+ torch_dtype=torch.bfloat16,
78
+ device_map="auto",
79
+ )
80
+ model.eval()
81
+
82
+ prompt = "Write a function to find the shared elements from the given two lists."
83
+ inputs = tokenizer.apply_chat_template(
84
+ [{'role': 'user', 'content': prompt }],
85
+ return_tensors="pt"
86
+ ).to(model.device)
87
+ outputs = model.generate(
88
+ inputs,
89
+ max_new_tokens=1024,
90
+ do_sample=False,
91
+ pad_token_id=tokenizer.eos_token_id,
92
+ eos_token_id=tokenizer.eos_token_id,
93
+ )
94
+ print(tokenizer.decode(outputs[0][len(inputs[0]):], skip_special_tokens=True))
95
+ ```
96
+
97
+
98
+ ## Contact
99
+
100
+ If you have any inquiries, please feel free to raise an issue or reach out to us via email at: xiangyue.work@gmail.com, zhengtianyu0428@gmail.com.
101
+ We're here to assist you!"