File size: 2,135 Bytes
209cc3c
 
 
 
 
 
 
b5971de
 
209cc3c
 
 
 
 
d14b349
209cc3c
b5971de
209cc3c
2600f98
 
209cc3c
 
 
b5971de
209cc3c
 
 
 
 
 
 
 
d14b349
 
209cc3c
 
b5971de
209cc3c
b5971de
 
 
209cc3c
 
 
 
20591ae
209cc3c
 
e7f35b9
209cc3c
 
 
2e49acd
20591ae
209cc3c
 
 
 
 
 
b5971de
209cc3c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
---
license: apache-2.0
base_model: Qwen/Qwen2.5-Coder-3B
tags:
- code
- humaneval
- multi-agent
- grpo
- multi-agent
- qwen2.5
library_name: transformers
pipeline_tag: text-generation
---

# 2xQwen2.5-Coder-3B-Cyclops-Main

This model is a fine-tuned version of 2 **Qwen/Qwen2.5-Coder-3B** using Multi-LLM Group Relative Policy Optimization (MLGRPO) on OpenAI HumanEval dataset.

"Cyclops" relies heavily on its `aux()` function for core implementation, while the main function adds edge case handling and refinements — just like a cyclops wielding power through its single eye.

## Model Details

- **Base Model**: Qwen/Qwen2.5-Coder-3B
- **Training Method**: MLGRPO (Multi-LLM Group Relative Policy Optimization)
- **Dataset**: HumanEval
- **Task**: Code generation with auxiliary function collaboration

## Usage

```python
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("LovelyBuggies/2xQwen2.5-Coder-3B-Cyclops-Main")
model = AutoModelForCausalLM.from_pretrained("LovelyBuggies/2xQwen2.5-Coder-3B-Cyclops-Main")

# Generate code
inputs = tokenizer(main_prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=256)
main_completion = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
cleaned_main_completion = extract_specific_function(cleanup_code(main_completion), example['entry_point'])
print(cleaned_main_completion)
```

## Training Details

This model was trained as part of a multi-LLM system on the full HumanEval dataset:
- Agent 0 generates auxiliary functions to help solve coding problems
- Agent 1 generates main functions that utilize the auxiliary functions
- Both agents are trained collaboratively using MLGRPO


## Agent Role  
- This is the **Main Function Generator** agent that creates the primary solution functions.
- It will identify edge cases first and then call auxiliary functions generated by [2xQwen2.5-Coder-3B-Cyclops-Aux](https://huggingface.co/LovelyBuggies/2xQwen2.5-Coder-3B-Cyclops-Aux/) to write code more effectively.


## Citation

If you use this model, please cite:
```bibtex
Coming soon.
```