File size: 2,435 Bytes
aa5bfd3 65e12cf aa5bfd3 65e12cf aa5bfd3 1d51917 aa5bfd3 1d51917 aa5bfd3 70daccf aa5bfd3 |
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 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 |
---
license: apache-2.0
tags:
- code-review
- multi-language
- mlx
- gguf
- qwen2.5-coder
base_model: Qwen/Qwen2.5-Coder-1.5B-Instruct
---
# AI Code Review Model
Multi-language code review model optimized for automated code review in CI/CD pipelines.
## Model Details
- **Base Model**: Qwen/Qwen2.5-Coder-1.5B-Instruct
- **Training Method**: LoRA fine-tuning with MLX
- **Format**: GGUF (Q4_K_M quantization)
- **Purpose**: Automated code review for CI/CD pipelines
## Usage
### Docker (Recommended)
```bash
docker pull ghcr.io/iq2i/ai-code-review:latest
# Review your codebase
docker run --rm -v $(pwd):/workspace ghcr.io/iq2i/ai-code-review:latest /workspace/src
```
### llama.cpp
```bash
# Download the model
wget https://huggingface.co/iq2i/ai-code-review/resolve/main/model-Q4_K_M.gguf
# Run inference
./llama-cli -m model-Q4_K_M.gguf -p "Review this code: ..."
```
### Python (llama-cpp-python)
```python
from llama_cpp import Llama
llm = Llama(model_path="model-Q4_K_M.gguf")
output = llm("Review this code: ...", max_tokens=512)
print(output)
```
## Output Format
The model outputs concise text-based code reviews:
```
**SQL injection vulnerability**
User input is concatenated directly into a raw SQL query without parameterization or escaping.
Impact: An attacker can execute arbitrary SQL commands, potentially dumping the entire database, deleting data, or escalating privileges. For example: keyword=' OR '1'='1' -- would return all products.
Suggestion:
Use parameter binding: DB::select("SELECT * FROM products WHERE name LIKE ?", ['%' . $keyword . '%']) or better, use Eloquent: Product::where('name', 'like', '%' . $keyword . '%')->get()
```
## Training
- **Training examples**: 100+ real-world code issues
- **Format**: ChatML conversation format with concise reviews
- **Framework**: MLX for Apple Silicon acceleration
- **Method**: LoRA adapters (r=4, alpha=8)
- **Iterations**: 625
For training details, see the [GitHub repository](https://github.com/iq2i/ai-code-review).
## Limitations
- Should be used as a supplementary tool, not a replacement for human review
- May not catch all edge cases or security vulnerabilities
- Best results on common programming patterns and frameworks
## License
Apache 2.0
## Citation
```bibtex
@software{ai_code_review,
title = {AI Code Review Model},
author = {IQ2i Team},
year = {2025},
url = {https://github.com/iq2i/ai-code-review}
}
```
|