初始化项目,由ModelHub XC社区提供模型
Model: CyrusCheungkf/git-commit-3B Source: Original Platform
This commit is contained in:
55
README.md
Normal file
55
README.md
Normal file
@@ -0,0 +1,55 @@
|
||||
---
|
||||
library_name: transformers
|
||||
tags:
|
||||
- git
|
||||
datasets:
|
||||
- Maxscha/commitbench
|
||||
language:
|
||||
- en
|
||||
base_model:
|
||||
- Qwen/Qwen2.5-Coder-3B-Instruct
|
||||
pipeline_tag: text-generation
|
||||
---
|
||||
|
||||
# Model Card for Model ID
|
||||
|
||||
Fine tuned Qwen2.5 3B model for writing git commit message. Used dataset Maxscha/commitbench
|
||||
|
||||
|
||||
## Model Details
|
||||
|
||||
- **Developed by:** Cyrus Cheung
|
||||
- **Model type:** Qwen2.5 3B
|
||||
- **License:** qwen-research
|
||||
- **Finetuned from model:** Qwen/Qwen2.5-Coder-3B-Instruct
|
||||
|
||||
## Uses
|
||||
|
||||
```python
|
||||
from transformers.models.auto.modeling_auto import AutoModelForCausalLM
|
||||
from transformers.models.auto.tokenization_auto import AutoTokenizer
|
||||
|
||||
model = AutoModelForCausalLM.from_pretrained("CyrusCheungkf/git-commit-3B")
|
||||
tokenizer = AutoTokenizer.from_pretrained("CyrusCheungkf/git-commit-3B")
|
||||
git_diff = "Output from using 'git diff'"
|
||||
|
||||
INSTRUCTION = """You are Git Commit Message Pro, a specialist in crafting precise, professional Git commit messages from .diff files. Your role is to analyze these files, interpret the changes, and generate a clear, direct commit message.
|
||||
|
||||
Guidelines:
|
||||
1. Be specific about the type of change (e.g., "Rename variable X to Y", "Extract method Z from class W").
|
||||
2. Prefer to write it on why and how instead of what changed.
|
||||
3. Interpret the changes; do not transcribe the diff.
|
||||
4. If you cannot read the entire file, attempt to generate a message based on the available information.
|
||||
5. Be concise and summarize the most important changes. Keep your response in 1 sentence."""
|
||||
conversation = [
|
||||
{"role": "user", "content": INSTRUCTION + "\n\nInputs:\n" + git_diff},
|
||||
]
|
||||
tokens = tokenizer.apply_chat_template(
|
||||
conversation, add_generation_prompt=True, return_tensors="pt", return_dict=True
|
||||
)
|
||||
output = model.generate(
|
||||
inputs=tokens["input_ids"],
|
||||
attention_mask=tokens["attention_mask"],
|
||||
)
|
||||
print(output)
|
||||
```
|
||||
Reference in New Issue
Block a user