Add README with usage instructions
Browse files
README.md
CHANGED
|
@@ -1,55 +1,45 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
language:
|
| 5 |
- en
|
|
|
|
| 6 |
pipeline_tag: text-generation
|
| 7 |
-
base_model: Qwen/Qwen2.5-0.5B
|
| 8 |
-
tags:
|
| 9 |
-
- chat
|
| 10 |
-
library_name: transformers
|
| 11 |
---
|
| 12 |
|
| 13 |
-
# Qwen2.5-0.5B-Instruct
|
| 14 |
-
|
| 15 |
-
## Introduction
|
| 16 |
|
| 17 |
-
|
| 18 |
|
| 19 |
-
|
| 20 |
-
- Significant improvements in **instruction following**, **generating long texts** (over 8K tokens), **understanding structured data** (e.g, tables), and **generating structured outputs** especially JSON. **More resilient to the diversity of system prompts**, enhancing role-play implementation and condition-setting for chatbots.
|
| 21 |
-
- **Long-context Support** up to 128K tokens and can generate up to 8K tokens.
|
| 22 |
-
- **Multilingual support** for over 29 languages, including Chinese, English, French, Spanish, Portuguese, German, Italian, Russian, Japanese, Korean, Vietnamese, Thai, Arabic, and more.
|
| 23 |
|
| 24 |
-
**
|
| 25 |
-
-
|
| 26 |
-
-
|
| 27 |
-
-
|
| 28 |
-
- Number of Parameters: 0.49B
|
| 29 |
-
- Number of Paramaters (Non-Embedding): 0.36B
|
| 30 |
-
- Number of Layers: 24
|
| 31 |
-
- Number of Attention Heads (GQA): 14 for Q and 2 for KV
|
| 32 |
-
- Context Length: Full 32,768 tokens and generation 8192 tokens
|
| 33 |
|
| 34 |
-
|
| 35 |
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
```
|
| 42 |
-
KeyError: 'qwen2'
|
| 43 |
-
```
|
| 44 |
|
| 45 |
-
##
|
| 46 |
|
| 47 |
-
|
| 48 |
|
| 49 |
```python
|
| 50 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 51 |
|
| 52 |
-
model_name = "
|
| 53 |
|
| 54 |
model = AutoModelForCausalLM.from_pretrained(
|
| 55 |
model_name,
|
|
@@ -58,53 +48,81 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 58 |
)
|
| 59 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 60 |
|
| 61 |
-
prompt = "
|
| 62 |
messages = [
|
| 63 |
-
{"role": "system", "content": "You are
|
| 64 |
{"role": "user", "content": prompt}
|
| 65 |
]
|
|
|
|
| 66 |
text = tokenizer.apply_chat_template(
|
| 67 |
messages,
|
| 68 |
tokenize=False,
|
| 69 |
add_generation_prompt=True
|
| 70 |
)
|
|
|
|
| 71 |
model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
|
| 72 |
|
| 73 |
generated_ids = model.generate(
|
| 74 |
**model_inputs,
|
| 75 |
max_new_tokens=512
|
| 76 |
)
|
| 77 |
-
generated_ids = [
|
| 78 |
-
output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
|
| 79 |
-
]
|
| 80 |
|
| 81 |
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
|
|
|
| 82 |
```
|
| 83 |
|
|
|
|
| 84 |
|
| 85 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 86 |
|
| 87 |
-
|
|
|
|
|
|
|
|
|
|
| 88 |
|
| 89 |
-
|
| 90 |
|
| 91 |
-
|
| 92 |
|
| 93 |
-
|
|
|
|
| 94 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 95 |
```
|
| 96 |
-
|
| 97 |
-
|
| 98 |
-
|
| 99 |
-
|
| 100 |
-
|
| 101 |
-
|
| 102 |
-
|
| 103 |
-
|
| 104 |
-
|
| 105 |
-
|
| 106 |
-
|
| 107 |
-
|
| 108 |
-
|
| 109 |
-
|
| 110 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
+
base_model: Qwen/Qwen2.5-0.5B-Instruct
|
| 4 |
+
tags:
|
| 5 |
+
- qwen
|
| 6 |
+
- qwen2.5
|
| 7 |
+
- instruct
|
| 8 |
+
- runpod
|
| 9 |
+
- serverless
|
| 10 |
language:
|
| 11 |
- en
|
| 12 |
+
- zh
|
| 13 |
pipeline_tag: text-generation
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
---
|
| 15 |
|
| 16 |
+
# Qwen2.5-0.5B-Instruct (Customizable Copy)
|
|
|
|
|
|
|
| 17 |
|
| 18 |
+
This is a copy of [Qwen/Qwen2.5-0.5B-Instruct](https://huggingface.co/Qwen/Qwen2.5-0.5B-Instruct) for customization and fine-tuning.
|
| 19 |
|
| 20 |
+
## π Model Details
|
|
|
|
|
|
|
|
|
|
| 21 |
|
| 22 |
+
- **Base Model:** Qwen/Qwen2.5-0.5B-Instruct
|
| 23 |
+
- **Size:** 0.5B parameters (~1GB)
|
| 24 |
+
- **Type:** Instruction-tuned language model
|
| 25 |
+
- **License:** Apache 2.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
|
| 27 |
+
## π― Purpose
|
| 28 |
|
| 29 |
+
This repository contains a **modifiable copy** of Qwen 2.5 for:
|
| 30 |
+
- Fine-tuning on custom datasets
|
| 31 |
+
- Experimentation and testing
|
| 32 |
+
- RunPod serverless deployment
|
| 33 |
+
- Model modifications
|
|
|
|
|
|
|
|
|
|
| 34 |
|
| 35 |
+
## π Usage
|
| 36 |
|
| 37 |
+
### Direct Inference
|
| 38 |
|
| 39 |
```python
|
| 40 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 41 |
|
| 42 |
+
model_name = "marcosremar2/runpod_serverless_n2"
|
| 43 |
|
| 44 |
model = AutoModelForCausalLM.from_pretrained(
|
| 45 |
model_name,
|
|
|
|
| 48 |
)
|
| 49 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 50 |
|
| 51 |
+
prompt = "What is artificial intelligence?"
|
| 52 |
messages = [
|
| 53 |
+
{"role": "system", "content": "You are a helpful assistant."},
|
| 54 |
{"role": "user", "content": prompt}
|
| 55 |
]
|
| 56 |
+
|
| 57 |
text = tokenizer.apply_chat_template(
|
| 58 |
messages,
|
| 59 |
tokenize=False,
|
| 60 |
add_generation_prompt=True
|
| 61 |
)
|
| 62 |
+
|
| 63 |
model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
|
| 64 |
|
| 65 |
generated_ids = model.generate(
|
| 66 |
**model_inputs,
|
| 67 |
max_new_tokens=512
|
| 68 |
)
|
|
|
|
|
|
|
|
|
|
| 69 |
|
| 70 |
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
| 71 |
+
print(response)
|
| 72 |
```
|
| 73 |
|
| 74 |
+
### RunPod Serverless Deployment
|
| 75 |
|
| 76 |
+
```yaml
|
| 77 |
+
Environment Variables:
|
| 78 |
+
MODEL_NAME: marcosremar2/runpod_serverless_n2
|
| 79 |
+
HF_TOKEN: YOUR_TOKEN_HERE
|
| 80 |
+
MAX_MODEL_LEN: 4096
|
| 81 |
+
TRUST_REMOTE_CODE: true
|
| 82 |
|
| 83 |
+
GPU: RTX 4090 (24GB)
|
| 84 |
+
Min Workers: 0
|
| 85 |
+
Max Workers: 1
|
| 86 |
+
```
|
| 87 |
|
| 88 |
+
## π§ Fine-tuning
|
| 89 |
|
| 90 |
+
To fine-tune this model:
|
| 91 |
|
| 92 |
+
```python
|
| 93 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer, TrainingArguments, Trainer
|
| 94 |
|
| 95 |
+
model = AutoModelForCausalLM.from_pretrained("marcosremar2/runpod_serverless_n2")
|
| 96 |
+
tokenizer = AutoTokenizer.from_pretrained("marcosremar2/runpod_serverless_n2")
|
| 97 |
+
|
| 98 |
+
# Your fine-tuning code here
|
| 99 |
+
# ...
|
| 100 |
+
|
| 101 |
+
# Push back to your repo
|
| 102 |
+
model.push_to_hub("marcosremar2/runpod_serverless_n2")
|
| 103 |
+
tokenizer.push_to_hub("marcosremar2/runpod_serverless_n2")
|
| 104 |
```
|
| 105 |
+
|
| 106 |
+
## π Performance
|
| 107 |
+
|
| 108 |
+
| Metric | Value |
|
| 109 |
+
|--------|-------|
|
| 110 |
+
| Parameters | 0.5B |
|
| 111 |
+
| Context Length | 32K tokens |
|
| 112 |
+
| VRAM Required | ~1-2GB |
|
| 113 |
+
| Inference Speed | 200-300 tokens/sec (RTX 4090) |
|
| 114 |
+
|
| 115 |
+
## π Original Model
|
| 116 |
+
|
| 117 |
+
This is based on: [Qwen/Qwen2.5-0.5B-Instruct](https://huggingface.co/Qwen/Qwen2.5-0.5B-Instruct)
|
| 118 |
+
|
| 119 |
+
For more information about the Qwen2.5 series, visit the original repository.
|
| 120 |
+
|
| 121 |
+
## π License
|
| 122 |
+
|
| 123 |
+
Apache 2.0 - Same as the original Qwen model.
|
| 124 |
+
|
| 125 |
+
## π Credits
|
| 126 |
+
|
| 127 |
+
- **Original Model:** Qwen Team @ Alibaba Cloud
|
| 128 |
+
- **Repository:** Custom copy for modification and deployment
|