Upload CAI-20B Marketing Strategy Expert model
Browse files- .gitattributes +1 -0
- DEPLOYMENT_INSTRUCTIONS.md +132 -0
- README.md +292 -5
- chat_template.jinja +397 -0
- config.json +76 -0
- generation_config.json +10 -0
- inference.py +183 -0
- model-00001-of-00009.safetensors +3 -0
- model-00002-of-00009.safetensors +3 -0
- model-00003-of-00009.safetensors +3 -0
- model-00004-of-00009.safetensors +3 -0
- model-00005-of-00009.safetensors +3 -0
- model-00006-of-00009.safetensors +3 -0
- model-00007-of-00009.safetensors +3 -0
- model-00008-of-00009.safetensors +3 -0
- model-00009-of-00009.safetensors +3 -0
- model.safetensors.index.json +419 -0
- model_card.json +20 -0
- special_tokens_map.json +23 -0
- tokenizer.json +3 -0
- tokenizer_config.json +183 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
DEPLOYMENT_INSTRUCTIONS.md
ADDED
|
@@ -0,0 +1,132 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# CAI-20B Deployment Instructions
|
| 2 |
+
|
| 3 |
+
## Prerequisites
|
| 4 |
+
|
| 5 |
+
1. **Hugging Face Account**: Create an account at https://huggingface.co
|
| 6 |
+
2. **Access Token**: Generate a token at https://huggingface.co/settings/tokens
|
| 7 |
+
3. **Git LFS**: Already installed in this environment
|
| 8 |
+
|
| 9 |
+
## Deployment Steps
|
| 10 |
+
|
| 11 |
+
### Option 1: Using Hugging Face CLI (Recommended)
|
| 12 |
+
|
| 13 |
+
```bash
|
| 14 |
+
# 1. Login to Hugging Face
|
| 15 |
+
huggingface-cli login
|
| 16 |
+
# Enter your token when prompted
|
| 17 |
+
|
| 18 |
+
# 2. Upload the model
|
| 19 |
+
python3 upload_to_hf.py
|
| 20 |
+
|
| 21 |
+
# Or manually:
|
| 22 |
+
huggingface-cli upload tigres2526/CAI-20B . --repo-type model
|
| 23 |
+
```
|
| 24 |
+
|
| 25 |
+
### Option 2: Using Git (Alternative)
|
| 26 |
+
|
| 27 |
+
```bash
|
| 28 |
+
# 1. Set up credentials
|
| 29 |
+
git config --global user.email "your-email@example.com"
|
| 30 |
+
git config --global user.name "Your Name"
|
| 31 |
+
|
| 32 |
+
# 2. Add remote
|
| 33 |
+
git remote add origin https://huggingface.co/tigres2526/CAI-20B
|
| 34 |
+
|
| 35 |
+
# 3. Commit and push
|
| 36 |
+
git add .
|
| 37 |
+
git commit -m "Upload CAI-20B Marketing Strategy Expert"
|
| 38 |
+
git push -u origin main
|
| 39 |
+
```
|
| 40 |
+
|
| 41 |
+
### Option 3: Using Python Script
|
| 42 |
+
|
| 43 |
+
```python
|
| 44 |
+
from huggingface_hub import HfApi, login
|
| 45 |
+
|
| 46 |
+
# Login
|
| 47 |
+
login(token="your-token-here")
|
| 48 |
+
|
| 49 |
+
# Upload
|
| 50 |
+
api = HfApi()
|
| 51 |
+
api.upload_folder(
|
| 52 |
+
folder_path=".",
|
| 53 |
+
repo_id="tigres2526/CAI-20B",
|
| 54 |
+
repo_type="model"
|
| 55 |
+
)
|
| 56 |
+
```
|
| 57 |
+
|
| 58 |
+
## Files to Deploy
|
| 59 |
+
|
| 60 |
+
The following files are ready for deployment:
|
| 61 |
+
|
| 62 |
+
```
|
| 63 |
+
✅ README.md # Model card with documentation
|
| 64 |
+
✅ config.json # Model configuration
|
| 65 |
+
✅ generation_config.json # Generation parameters
|
| 66 |
+
✅ tokenizer.json # Tokenizer
|
| 67 |
+
✅ tokenizer_config.json # Tokenizer config
|
| 68 |
+
✅ special_tokens_map.json # Special tokens
|
| 69 |
+
✅ chat_template.jinja # Chat template
|
| 70 |
+
✅ model-*.safetensors # Model weights (9 files, 39GB total)
|
| 71 |
+
✅ model.safetensors.index.json # Model index
|
| 72 |
+
✅ inference.py # Example inference script
|
| 73 |
+
```
|
| 74 |
+
|
| 75 |
+
## After Deployment
|
| 76 |
+
|
| 77 |
+
### 1. Verify Upload
|
| 78 |
+
Visit: https://huggingface.co/tigres2526/CAI-20B
|
| 79 |
+
|
| 80 |
+
### 2. Test the Model
|
| 81 |
+
```python
|
| 82 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 83 |
+
|
| 84 |
+
model = AutoModelForCausalLM.from_pretrained("tigres2526/CAI-20B")
|
| 85 |
+
tokenizer = AutoTokenizer.from_pretrained("tigres2526/CAI-20B")
|
| 86 |
+
|
| 87 |
+
# Test generation
|
| 88 |
+
inputs = tokenizer("What are the best marketing channels?", return_tensors="pt")
|
| 89 |
+
outputs = model.generate(**inputs, max_new_tokens=100)
|
| 90 |
+
print(tokenizer.decode(outputs[0]))
|
| 91 |
+
```
|
| 92 |
+
|
| 93 |
+
### 3. Enable Inference API
|
| 94 |
+
Go to model settings and enable the Inference API for quick testing.
|
| 95 |
+
|
| 96 |
+
## Model Information
|
| 97 |
+
|
| 98 |
+
- **Size**: 39GB (20B parameters)
|
| 99 |
+
- **Format**: Safetensors
|
| 100 |
+
- **Base Model**: openai/gpt-oss-20b
|
| 101 |
+
- **Fine-tuning**: Marketing strategy expertise
|
| 102 |
+
- **Performance**: 79.5% benchmark score
|
| 103 |
+
|
| 104 |
+
## Troubleshooting
|
| 105 |
+
|
| 106 |
+
### Authentication Issues
|
| 107 |
+
```bash
|
| 108 |
+
# Clear credentials and re-login
|
| 109 |
+
huggingface-cli logout
|
| 110 |
+
huggingface-cli login
|
| 111 |
+
```
|
| 112 |
+
|
| 113 |
+
### Upload Failures
|
| 114 |
+
- Check internet connection
|
| 115 |
+
- Verify repository permissions
|
| 116 |
+
- Use smaller chunk sizes for large files
|
| 117 |
+
|
| 118 |
+
### Out of Space
|
| 119 |
+
The model is 39GB. Ensure sufficient disk space and bandwidth.
|
| 120 |
+
|
| 121 |
+
## Support
|
| 122 |
+
|
| 123 |
+
- Model issues: Open an issue on the HF repository
|
| 124 |
+
- Upload help: https://huggingface.co/docs/hub/upload
|
| 125 |
+
- Community: https://discuss.huggingface.co
|
| 126 |
+
|
| 127 |
+
## Notes
|
| 128 |
+
|
| 129 |
+
- The model includes response cleanup utilities for production use
|
| 130 |
+
- Optimal temperature: 0.7
|
| 131 |
+
- Recommended max_tokens: 250
|
| 132 |
+
- Use repetition_penalty: 1.1 to avoid repetitive text
|
README.md
CHANGED
|
@@ -1,5 +1,292 @@
|
|
| 1 |
-
---
|
| 2 |
-
license:
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
language:
|
| 4 |
+
- en
|
| 5 |
+
library_name: transformers
|
| 6 |
+
tags:
|
| 7 |
+
- marketing
|
| 8 |
+
- business
|
| 9 |
+
- strategy
|
| 10 |
+
- conversational
|
| 11 |
+
- gpt-oss
|
| 12 |
+
- fine-tuned
|
| 13 |
+
datasets:
|
| 14 |
+
- custom-marketing-dataset
|
| 15 |
+
model_name: CAI-20B Marketing Strategy Expert
|
| 16 |
+
base_model: openai/gpt-oss-20b
|
| 17 |
+
inference: true
|
| 18 |
+
widget:
|
| 19 |
+
- text: "What are the best marketing channels for a B2B SaaS startup?"
|
| 20 |
+
example_title: "Marketing Channels"
|
| 21 |
+
- text: "How should I allocate a $10K monthly marketing budget?"
|
| 22 |
+
example_title: "Budget Allocation"
|
| 23 |
+
- text: "What's the difference between CAC and LTV?"
|
| 24 |
+
example_title: "Marketing Metrics"
|
| 25 |
+
---
|
| 26 |
+
|
| 27 |
+
# CAI-20B: Marketing Strategy Expert
|
| 28 |
+
|
| 29 |
+
A fine-tuned version of OpenAI's GPT-OSS-20B model specialized for marketing strategy, performance marketing, and business growth advice.
|
| 30 |
+
|
| 31 |
+
## Model Details
|
| 32 |
+
|
| 33 |
+
### Model Description
|
| 34 |
+
|
| 35 |
+
CAI-20B is a 20-billion parameter language model fine-tuned on high-quality marketing strategy conversations. It excels at providing actionable marketing advice, campaign strategies, budget allocation recommendations, and growth tactics for businesses of all sizes.
|
| 36 |
+
|
| 37 |
+
- **Developed by:** tigres2526
|
| 38 |
+
- **Model type:** Causal Language Model (Fine-tuned)
|
| 39 |
+
- **Language(s):** English
|
| 40 |
+
- **License:** Apache 2.0
|
| 41 |
+
- **Finetuned from:** openai/gpt-oss-20b
|
| 42 |
+
|
| 43 |
+
### Model Performance
|
| 44 |
+
|
| 45 |
+
Overall Benchmark Score: **79.5%**
|
| 46 |
+
|
| 47 |
+
#### Category Performance:
|
| 48 |
+
- 🎯 **Performance Marketing:** 100%
|
| 49 |
+
- 🏆 **Brand Positioning:** 100%
|
| 50 |
+
- 📊 **Data & Analytics:** 94%
|
| 51 |
+
- 📱 **Channel Expertise:** 79%
|
| 52 |
+
- 🧠 **Customer Psychology:** 64%
|
| 53 |
+
- 📝 **Content Strategy:** 64%
|
| 54 |
+
- 📈 **Strategic Planning:** 56%
|
| 55 |
+
|
| 56 |
+
## Uses
|
| 57 |
+
|
| 58 |
+
### Direct Use
|
| 59 |
+
|
| 60 |
+
This model is designed for:
|
| 61 |
+
- Marketing strategy consultation
|
| 62 |
+
- Campaign planning and optimization
|
| 63 |
+
- Budget allocation recommendations
|
| 64 |
+
- Channel selection and optimization
|
| 65 |
+
- Customer acquisition strategies
|
| 66 |
+
- Brand positioning advice
|
| 67 |
+
- Content marketing strategies
|
| 68 |
+
- Performance marketing optimization
|
| 69 |
+
|
| 70 |
+
### Recommended Use Cases
|
| 71 |
+
|
| 72 |
+
1. **Marketing Teams:** Get instant strategic advice for campaigns
|
| 73 |
+
2. **Startups:** Receive guidance on initial marketing strategies
|
| 74 |
+
3. **Consultants:** Augment expertise with data-driven insights
|
| 75 |
+
4. **Educators:** Teaching marketing concepts with practical examples
|
| 76 |
+
|
| 77 |
+
### Out-of-Scope Use
|
| 78 |
+
|
| 79 |
+
This model should NOT be used for:
|
| 80 |
+
- Medical, legal, or financial advice
|
| 81 |
+
- Generating misleading or deceptive content
|
| 82 |
+
- Making final business decisions without human review
|
| 83 |
+
- Personal data processing or storage
|
| 84 |
+
|
| 85 |
+
## Bias, Risks, and Limitations
|
| 86 |
+
|
| 87 |
+
### Known Limitations
|
| 88 |
+
|
| 89 |
+
1. **Response Artifacts:** ~25% of responses may contain minor formatting artifacts that require cleanup
|
| 90 |
+
2. **Context Length:** Optimal performance with inputs under 2048 tokens
|
| 91 |
+
3. **Strategic Planning:** Weaker performance on complex multi-year strategic planning (56% accuracy)
|
| 92 |
+
4. **Knowledge Cutoff:** Training data extends only to 2024-06
|
| 93 |
+
|
| 94 |
+
### Recommendations
|
| 95 |
+
|
| 96 |
+
- Always review outputs for accuracy and relevance
|
| 97 |
+
- Use the provided cleanup wrapper for production deployments
|
| 98 |
+
- Implement response validation for critical use cases
|
| 99 |
+
- Monitor response quality and collect user feedback
|
| 100 |
+
|
| 101 |
+
## How to Get Started with the Model
|
| 102 |
+
|
| 103 |
+
### Installation
|
| 104 |
+
|
| 105 |
+
```bash
|
| 106 |
+
pip install transformers torch peft accelerate
|
| 107 |
+
```
|
| 108 |
+
|
| 109 |
+
### Quick Start
|
| 110 |
+
|
| 111 |
+
```python
|
| 112 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 113 |
+
import torch
|
| 114 |
+
|
| 115 |
+
# Load model
|
| 116 |
+
model = AutoModelForCausalLM.from_pretrained(
|
| 117 |
+
"tigres2526/CAI-20B",
|
| 118 |
+
device_map="auto",
|
| 119 |
+
torch_dtype=torch.bfloat16,
|
| 120 |
+
trust_remote_code=True
|
| 121 |
+
)
|
| 122 |
+
|
| 123 |
+
tokenizer = AutoTokenizer.from_pretrained(
|
| 124 |
+
"tigres2526/CAI-20B",
|
| 125 |
+
trust_remote_code=True
|
| 126 |
+
)
|
| 127 |
+
|
| 128 |
+
# Generate response
|
| 129 |
+
def get_marketing_advice(question):
|
| 130 |
+
prompt = f"""You are a marketing strategy expert. Provide actionable advice.
|
| 131 |
+
|
| 132 |
+
User: {question}
|
| 133 |
+
Assistant:"""
|
| 134 |
+
|
| 135 |
+
inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=2048)
|
| 136 |
+
|
| 137 |
+
with torch.no_grad():
|
| 138 |
+
outputs = model.generate(
|
| 139 |
+
**inputs,
|
| 140 |
+
max_new_tokens=250,
|
| 141 |
+
temperature=0.7,
|
| 142 |
+
top_p=0.9,
|
| 143 |
+
repetition_penalty=1.1,
|
| 144 |
+
do_sample=True,
|
| 145 |
+
pad_token_id=tokenizer.pad_token_id
|
| 146 |
+
)
|
| 147 |
+
|
| 148 |
+
response = tokenizer.decode(outputs[0][inputs['input_ids'].shape[1]:], skip_special_tokens=True)
|
| 149 |
+
return response
|
| 150 |
+
|
| 151 |
+
# Example usage
|
| 152 |
+
advice = get_marketing_advice("How should I spend $10K on marketing for my SaaS startup?")
|
| 153 |
+
print(advice)
|
| 154 |
+
```
|
| 155 |
+
|
| 156 |
+
### Production Deployment with Cleanup
|
| 157 |
+
|
| 158 |
+
For production use, we recommend using our cleanup wrapper to ensure high-quality responses:
|
| 159 |
+
|
| 160 |
+
```python
|
| 161 |
+
import re
|
| 162 |
+
|
| 163 |
+
class ResponseCleaner:
|
| 164 |
+
def __init__(self):
|
| 165 |
+
self.artifact_patterns = [
|
| 166 |
+
r'<\|[^>]+\|>',
|
| 167 |
+
r'assistantfinal',
|
| 168 |
+
r'We need to.*?(?=\n|$)',
|
| 169 |
+
r'Let me.*?(?=\n|$)',
|
| 170 |
+
]
|
| 171 |
+
|
| 172 |
+
def clean_response(self, text):
|
| 173 |
+
cleaned = text
|
| 174 |
+
for pattern in self.artifact_patterns:
|
| 175 |
+
cleaned = re.sub(pattern, '', cleaned, flags=re.IGNORECASE)
|
| 176 |
+
|
| 177 |
+
# Remove multiple spaces and trailing content
|
| 178 |
+
cleaned = re.sub(r'\s+', ' ', cleaned).strip()
|
| 179 |
+
|
| 180 |
+
# Ensure proper ending
|
| 181 |
+
if cleaned and cleaned[-1] not in '.!?':
|
| 182 |
+
last_sentence = cleaned.split('.')[-1].strip()
|
| 183 |
+
if len(last_sentence) < 20:
|
| 184 |
+
parts = cleaned.rsplit('.', 1)
|
| 185 |
+
if len(parts) > 1:
|
| 186 |
+
cleaned = parts[0] + '.'
|
| 187 |
+
else:
|
| 188 |
+
cleaned += '.'
|
| 189 |
+
|
| 190 |
+
return cleaned
|
| 191 |
+
|
| 192 |
+
# Usage
|
| 193 |
+
cleaner = ResponseCleaner()
|
| 194 |
+
raw_response = model.generate(...)
|
| 195 |
+
clean_response = cleaner.clean_response(raw_response)
|
| 196 |
+
```
|
| 197 |
+
|
| 198 |
+
## Training Details
|
| 199 |
+
|
| 200 |
+
### Training Data
|
| 201 |
+
|
| 202 |
+
The model was fine-tuned on:
|
| 203 |
+
- 1,000+ curated marketing strategy conversations
|
| 204 |
+
- 100+ real-world marketing scenarios
|
| 205 |
+
- 50+ preference optimization pairs
|
| 206 |
+
- Topics covering all major marketing domains
|
| 207 |
+
|
| 208 |
+
### Training Procedure
|
| 209 |
+
|
| 210 |
+
#### Training Hyperparameters
|
| 211 |
+
|
| 212 |
+
- **Training regime:** QLoRA with 4-bit quantization
|
| 213 |
+
- **LoRA Rank:** 32
|
| 214 |
+
- **LoRA Alpha:** 64
|
| 215 |
+
- **Learning Rate:** 1e-5 (SFT), 5e-6 (DPO)
|
| 216 |
+
- **Batch Size:** 4
|
| 217 |
+
- **Epochs:** 4 (SFT) + 2 (DPO)
|
| 218 |
+
- **Optimizer:** Paged AdamW 32-bit
|
| 219 |
+
|
| 220 |
+
#### Hardware
|
| 221 |
+
|
| 222 |
+
- **GPU:** NVIDIA H100 80GB
|
| 223 |
+
- **Training Time:** ~18 hours total
|
| 224 |
+
- **Framework:** PyTorch 2.0 with Transformers 4.40+
|
| 225 |
+
|
| 226 |
+
## Evaluation
|
| 227 |
+
|
| 228 |
+
### Testing Methodology
|
| 229 |
+
|
| 230 |
+
Evaluated on 200+ marketing strategy questions across 7 categories:
|
| 231 |
+
- Performance Marketing
|
| 232 |
+
- Brand Positioning
|
| 233 |
+
- Strategic Planning
|
| 234 |
+
- Content Strategy
|
| 235 |
+
- Customer Psychology
|
| 236 |
+
- Data & Analytics
|
| 237 |
+
- Channel Expertise
|
| 238 |
+
|
| 239 |
+
### Metrics
|
| 240 |
+
|
| 241 |
+
- **Overall Accuracy:** 79.5%
|
| 242 |
+
- **Response Coherence:** 85%
|
| 243 |
+
- **Actionability:** 82%
|
| 244 |
+
- **Technical Accuracy:** 88%
|
| 245 |
+
|
| 246 |
+
## Environmental Impact
|
| 247 |
+
|
| 248 |
+
- **Hardware Type:** NVIDIA H100
|
| 249 |
+
- **Hours used:** ~18
|
| 250 |
+
- **Carbon Emitted:** Estimated 2.7 kg CO2eq
|
| 251 |
+
|
| 252 |
+
## Technical Specifications
|
| 253 |
+
|
| 254 |
+
### Model Architecture
|
| 255 |
+
|
| 256 |
+
- **Base Model:** GPT-OSS-20B
|
| 257 |
+
- **Parameters:** 20 billion
|
| 258 |
+
- **Context Length:** 128K (optimal: 2-4K)
|
| 259 |
+
- **Vocabulary Size:** 200K (o200k_harmony tokenizer)
|
| 260 |
+
|
| 261 |
+
### Compute Infrastructure
|
| 262 |
+
|
| 263 |
+
- Single H100 80GB GPU
|
| 264 |
+
- Ubuntu 22.04
|
| 265 |
+
- CUDA 12.1
|
| 266 |
+
- PyTorch 2.0
|
| 267 |
+
|
| 268 |
+
## Citation
|
| 269 |
+
|
| 270 |
+
If you use this model, please cite:
|
| 271 |
+
|
| 272 |
+
```bibtex
|
| 273 |
+
@misc{cai20b2025,
|
| 274 |
+
title={CAI-20B: Marketing Strategy Expert},
|
| 275 |
+
author={tigres2526},
|
| 276 |
+
year={2025},
|
| 277 |
+
publisher={Hugging Face},
|
| 278 |
+
howpublished={\url{https://huggingface.co/tigres2526/CAI-20B}}
|
| 279 |
+
}
|
| 280 |
+
```
|
| 281 |
+
|
| 282 |
+
## Model Card Authors
|
| 283 |
+
|
| 284 |
+
tigres2526
|
| 285 |
+
|
| 286 |
+
## Model Card Contact
|
| 287 |
+
|
| 288 |
+
Please open an issue on the Hugging Face repository for questions or feedback.
|
| 289 |
+
|
| 290 |
+
## Disclaimer
|
| 291 |
+
|
| 292 |
+
This model is provided "as is" without warranties. Users should validate outputs for their specific use cases. Not intended to replace professional marketing consultants.
|
chat_template.jinja
ADDED
|
@@ -0,0 +1,397 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{#-
|
| 2 |
+
In addition to the normal inputs of `messages` and `tools`, this template also accepts the
|
| 3 |
+
following kwargs:
|
| 4 |
+
- "builtin_tools": A list, can contain "browser" and/or "python".
|
| 5 |
+
- "model_identity": A string that optionally describes the model identity.
|
| 6 |
+
- "reasoning_effort": A string that describes the reasoning effort, defaults to "medium".
|
| 7 |
+
#}
|
| 8 |
+
|
| 9 |
+
{#- Tool Definition Rendering ============================================== #}
|
| 10 |
+
{%- macro render_typescript_type(param_spec, required_params, is_nullable=false) -%}
|
| 11 |
+
{%- if param_spec.type == "array" -%}
|
| 12 |
+
{%- if param_spec['items'] -%}
|
| 13 |
+
{%- if param_spec['items']['type'] == "string" -%}
|
| 14 |
+
{{- "string[]" }}
|
| 15 |
+
{%- elif param_spec['items']['type'] == "number" -%}
|
| 16 |
+
{{- "number[]" }}
|
| 17 |
+
{%- elif param_spec['items']['type'] == "integer" -%}
|
| 18 |
+
{{- "number[]" }}
|
| 19 |
+
{%- elif param_spec['items']['type'] == "boolean" -%}
|
| 20 |
+
{{- "boolean[]" }}
|
| 21 |
+
{%- else -%}
|
| 22 |
+
{%- set inner_type = render_typescript_type(param_spec['items'], required_params) -%}
|
| 23 |
+
{%- if inner_type == "object | object" or inner_type|length > 50 -%}
|
| 24 |
+
{{- "any[]" }}
|
| 25 |
+
{%- else -%}
|
| 26 |
+
{{- inner_type + "[]" }}
|
| 27 |
+
{%- endif -%}
|
| 28 |
+
{%- endif -%}
|
| 29 |
+
{%- if param_spec.nullable -%}
|
| 30 |
+
{{- " | null" }}
|
| 31 |
+
{%- endif -%}
|
| 32 |
+
{%- else -%}
|
| 33 |
+
{{- "any[]" }}
|
| 34 |
+
{%- if param_spec.nullable -%}
|
| 35 |
+
{{- " | null" }}
|
| 36 |
+
{%- endif -%}
|
| 37 |
+
{%- endif -%}
|
| 38 |
+
{%- elif param_spec.type is defined and param_spec.type is iterable and param_spec.type is not string and param_spec.type is not mapping and param_spec.type[0] is defined -%}
|
| 39 |
+
{#- Handle array of types like ["object", "object"] from Union[dict, list] #}
|
| 40 |
+
{%- if param_spec.type | length > 1 -%}
|
| 41 |
+
{{- param_spec.type | join(" | ") }}
|
| 42 |
+
{%- else -%}
|
| 43 |
+
{{- param_spec.type[0] }}
|
| 44 |
+
{%- endif -%}
|
| 45 |
+
{%- elif param_spec.oneOf -%}
|
| 46 |
+
{#- Handle oneOf schemas - check for complex unions and fallback to any #}
|
| 47 |
+
{%- set has_object_variants = false -%}
|
| 48 |
+
{%- for variant in param_spec.oneOf -%}
|
| 49 |
+
{%- if variant.type == "object" -%}
|
| 50 |
+
{%- set has_object_variants = true -%}
|
| 51 |
+
{%- endif -%}
|
| 52 |
+
{%- endfor -%}
|
| 53 |
+
{%- if has_object_variants and param_spec.oneOf|length > 1 -%}
|
| 54 |
+
{{- "any" }}
|
| 55 |
+
{%- else -%}
|
| 56 |
+
{%- for variant in param_spec.oneOf -%}
|
| 57 |
+
{{- render_typescript_type(variant, required_params) -}}
|
| 58 |
+
{%- if variant.description %}
|
| 59 |
+
{{- "// " + variant.description }}
|
| 60 |
+
{%- endif -%}
|
| 61 |
+
{%- if variant.default is defined %}
|
| 62 |
+
{{ "// default: " + variant.default|tojson }}
|
| 63 |
+
{%- endif -%}
|
| 64 |
+
{%- if not loop.last %}
|
| 65 |
+
{{- " | " }}
|
| 66 |
+
{% endif -%}
|
| 67 |
+
{%- endfor -%}
|
| 68 |
+
{%- endif -%}
|
| 69 |
+
{%- elif param_spec.type == "string" -%}
|
| 70 |
+
{%- if param_spec.enum -%}
|
| 71 |
+
{{- '"' + param_spec.enum|join('" | "') + '"' -}}
|
| 72 |
+
{%- else -%}
|
| 73 |
+
{{- "string" }}
|
| 74 |
+
{%- if param_spec.nullable %}
|
| 75 |
+
{{- " | null" }}
|
| 76 |
+
{%- endif -%}
|
| 77 |
+
{%- endif -%}
|
| 78 |
+
{%- elif param_spec.type == "number" -%}
|
| 79 |
+
{{- "number" }}
|
| 80 |
+
{%- elif param_spec.type == "integer" -%}
|
| 81 |
+
{{- "number" }}
|
| 82 |
+
{%- elif param_spec.type == "boolean" -%}
|
| 83 |
+
{{- "boolean" }}
|
| 84 |
+
|
| 85 |
+
{%- elif param_spec.type == "object" -%}
|
| 86 |
+
{%- if param_spec.properties -%}
|
| 87 |
+
{{- "{
|
| 88 |
+
" }}
|
| 89 |
+
{%- for prop_name, prop_spec in param_spec.properties.items() -%}
|
| 90 |
+
{{- prop_name -}}
|
| 91 |
+
{%- if prop_name not in (param_spec.required or []) -%}
|
| 92 |
+
{{- "?" }}
|
| 93 |
+
{%- endif -%}
|
| 94 |
+
{{- ": " }}
|
| 95 |
+
{{ render_typescript_type(prop_spec, param_spec.required or []) }}
|
| 96 |
+
{%- if not loop.last -%}
|
| 97 |
+
{{-", " }}
|
| 98 |
+
{%- endif -%}
|
| 99 |
+
{%- endfor -%}
|
| 100 |
+
{{- "}" }}
|
| 101 |
+
{%- else -%}
|
| 102 |
+
{{- "object" }}
|
| 103 |
+
{%- endif -%}
|
| 104 |
+
{%- else -%}
|
| 105 |
+
{{- "any" }}
|
| 106 |
+
{%- endif -%}
|
| 107 |
+
{%- endmacro -%}
|
| 108 |
+
|
| 109 |
+
{%- macro render_tool_namespace(namespace_name, tools) -%}
|
| 110 |
+
{{- "## " + namespace_name + "
|
| 111 |
+
|
| 112 |
+
" }}
|
| 113 |
+
{{- "namespace " + namespace_name + " {
|
| 114 |
+
|
| 115 |
+
" }}
|
| 116 |
+
{%- for tool in tools %}
|
| 117 |
+
{%- set tool = tool.function %}
|
| 118 |
+
{{- "// " + tool.description + "
|
| 119 |
+
" }}
|
| 120 |
+
{{- "type "+ tool.name + " = " }}
|
| 121 |
+
{%- if tool.parameters and tool.parameters.properties %}
|
| 122 |
+
{{- "(_: {
|
| 123 |
+
" }}
|
| 124 |
+
{%- for param_name, param_spec in tool.parameters.properties.items() %}
|
| 125 |
+
{%- if param_spec.description %}
|
| 126 |
+
{{- "// " + param_spec.description + "
|
| 127 |
+
" }}
|
| 128 |
+
{%- endif %}
|
| 129 |
+
{{- param_name }}
|
| 130 |
+
{%- if param_name not in (tool.parameters.required or []) -%}
|
| 131 |
+
{{- "?" }}
|
| 132 |
+
{%- endif -%}
|
| 133 |
+
{{- ": " }}
|
| 134 |
+
{{- render_typescript_type(param_spec, tool.parameters.required or []) }}
|
| 135 |
+
{%- if param_spec.default is defined -%}
|
| 136 |
+
{%- if param_spec.enum %}
|
| 137 |
+
{{- ", // default: " + param_spec.default }}
|
| 138 |
+
{%- elif param_spec.oneOf %}
|
| 139 |
+
{{- "// default: " + param_spec.default }}
|
| 140 |
+
{%- else %}
|
| 141 |
+
{{- ", // default: " + param_spec.default|tojson }}
|
| 142 |
+
{%- endif -%}
|
| 143 |
+
{%- endif -%}
|
| 144 |
+
{%- if not loop.last %}
|
| 145 |
+
{{- ",
|
| 146 |
+
" }}
|
| 147 |
+
{%- else %}
|
| 148 |
+
{{- "
|
| 149 |
+
" }}
|
| 150 |
+
{%- endif -%}
|
| 151 |
+
{%- endfor %}
|
| 152 |
+
{{- "}) => any;
|
| 153 |
+
|
| 154 |
+
" }}
|
| 155 |
+
{%- else -%}
|
| 156 |
+
{{- "() => any;
|
| 157 |
+
|
| 158 |
+
" }}
|
| 159 |
+
{%- endif -%}
|
| 160 |
+
{%- endfor %}
|
| 161 |
+
{{- "} // namespace " + namespace_name }}
|
| 162 |
+
{%- endmacro -%}
|
| 163 |
+
|
| 164 |
+
{%- macro render_builtin_tools(browser_tool, python_tool) -%}
|
| 165 |
+
{%- if browser_tool %}
|
| 166 |
+
{{- "## browser
|
| 167 |
+
|
| 168 |
+
" }}
|
| 169 |
+
{{- "// Tool for browsing.
|
| 170 |
+
" }}
|
| 171 |
+
{{- "// The `cursor` appears in brackets before each browsing display: `[{cursor}]`.
|
| 172 |
+
" }}
|
| 173 |
+
{{- "// Cite information from the tool using the following format:
|
| 174 |
+
" }}
|
| 175 |
+
{{- "// `【{cursor}†L{line_start}(-L{line_end})?】`, for example: `【6†L9-L11】` or `【8†L3】`.
|
| 176 |
+
" }}
|
| 177 |
+
{{- "// Do not quote more than 10 words directly from the tool output.
|
| 178 |
+
" }}
|
| 179 |
+
{{- "// sources=web (default: web)
|
| 180 |
+
" }}
|
| 181 |
+
{{- "namespace browser {
|
| 182 |
+
|
| 183 |
+
" }}
|
| 184 |
+
{{- "// Searches for information related to `query` and displays `topn` results.
|
| 185 |
+
" }}
|
| 186 |
+
{{- "type search = (_: {
|
| 187 |
+
" }}
|
| 188 |
+
{{- "query: string,
|
| 189 |
+
" }}
|
| 190 |
+
{{- "topn?: number, // default: 10
|
| 191 |
+
" }}
|
| 192 |
+
{{- "source?: string,
|
| 193 |
+
" }}
|
| 194 |
+
{{- "}) => any;
|
| 195 |
+
|
| 196 |
+
" }}
|
| 197 |
+
{{- "// Opens the link `id` from the page indicated by `cursor` starting at line number `loc`, showing `num_lines` lines.
|
| 198 |
+
" }}
|
| 199 |
+
{{- "// Valid link ids are displayed with the formatting: `【{id}†.*】`.
|
| 200 |
+
" }}
|
| 201 |
+
{{- "// If `cursor` is not provided, the most recent page is implied.
|
| 202 |
+
" }}
|
| 203 |
+
{{- "// If `id` is a string, it is treated as a fully qualified URL associated with `source`.
|
| 204 |
+
" }}
|
| 205 |
+
{{- "// If `loc` is not provided, the viewport will be positioned at the beginning of the document or centered on the most relevant passage, if available.
|
| 206 |
+
" }}
|
| 207 |
+
{{- "// Use this function without `id` to scroll to a new location of an opened page.
|
| 208 |
+
" }}
|
| 209 |
+
{{- "type open = (_: {
|
| 210 |
+
" }}
|
| 211 |
+
{{- "id?: number | string, // default: -1
|
| 212 |
+
" }}
|
| 213 |
+
{{- "cursor?: number, // default: -1
|
| 214 |
+
" }}
|
| 215 |
+
{{- "loc?: number, // default: -1
|
| 216 |
+
" }}
|
| 217 |
+
{{- "num_lines?: number, // default: -1
|
| 218 |
+
" }}
|
| 219 |
+
{{- "view_source?: boolean, // default: false
|
| 220 |
+
" }}
|
| 221 |
+
{{- "source?: string,
|
| 222 |
+
" }}
|
| 223 |
+
{{- "}) => any;
|
| 224 |
+
|
| 225 |
+
" }}
|
| 226 |
+
{{- "// Finds exact matches of `pattern` in the current page, or the page given by `cursor`.
|
| 227 |
+
" }}
|
| 228 |
+
{{- "type find = (_: {
|
| 229 |
+
" }}
|
| 230 |
+
{{- "pattern: string,
|
| 231 |
+
" }}
|
| 232 |
+
{{- "cursor?: number, // default: -1
|
| 233 |
+
" }}
|
| 234 |
+
{{- "}) => any;
|
| 235 |
+
|
| 236 |
+
" }}
|
| 237 |
+
{{- "} // namespace browser
|
| 238 |
+
|
| 239 |
+
" }}
|
| 240 |
+
{%- endif -%}
|
| 241 |
+
|
| 242 |
+
{%- if python_tool %}
|
| 243 |
+
{{- "## python
|
| 244 |
+
|
| 245 |
+
" }}
|
| 246 |
+
{{- "Use this tool to execute Python code in your chain of thought. The code will not be shown to the user. This tool should be used for internal reasoning, but not for code that is intended to be visible to the user (e.g. when creating plots, tables, or files).
|
| 247 |
+
|
| 248 |
+
" }}
|
| 249 |
+
{{- "When you send a message containing Python code to python, it will be executed in a stateful Jupyter notebook environment. python will respond with the output of the execution or time out after 120.0 seconds. The drive at '/mnt/data' can be used to save and persist user files. Internet access for this session is UNKNOWN. Depends on the cluster.
|
| 250 |
+
|
| 251 |
+
" }}
|
| 252 |
+
{%- endif -%}
|
| 253 |
+
{%- endmacro -%}
|
| 254 |
+
|
| 255 |
+
{#- System Message Construction ============================================ #}
|
| 256 |
+
{%- macro build_system_message() -%}
|
| 257 |
+
{%- if model_identity is not defined %}
|
| 258 |
+
{%- set model_identity = "You are ChatGPT, a large language model trained by OpenAI." %}
|
| 259 |
+
{%- endif %}
|
| 260 |
+
{{- model_identity + "
|
| 261 |
+
" }}
|
| 262 |
+
{{- "Knowledge cutoff: 2024-06
|
| 263 |
+
" }}
|
| 264 |
+
{{- "Current date: " + strftime_now("%Y-%m-%d") + "
|
| 265 |
+
|
| 266 |
+
" }}
|
| 267 |
+
{%- if reasoning_effort is not defined %}
|
| 268 |
+
{%- set reasoning_effort = "medium" %}
|
| 269 |
+
{%- endif %}
|
| 270 |
+
{{- "Reasoning: " + reasoning_effort + "
|
| 271 |
+
|
| 272 |
+
" }}
|
| 273 |
+
{%- if builtin_tools %}
|
| 274 |
+
{{- "# Tools
|
| 275 |
+
|
| 276 |
+
" }}
|
| 277 |
+
{%- set available_builtin_tools = namespace(browser=false, python=false) %}
|
| 278 |
+
{%- for tool in builtin_tools %}
|
| 279 |
+
{%- if tool == "browser" %}
|
| 280 |
+
{%- set available_builtin_tools.browser = true %}
|
| 281 |
+
{%- elif tool == "python" %}
|
| 282 |
+
{%- set available_builtin_tools.python = true %}
|
| 283 |
+
{%- endif %}
|
| 284 |
+
{%- endfor %}
|
| 285 |
+
{{- render_builtin_tools(available_builtin_tools.browser, available_builtin_tools.python) }}
|
| 286 |
+
{%- endif -%}
|
| 287 |
+
{{- "# Valid channels: analysis, commentary, final. Channel must be included for every message." }}
|
| 288 |
+
{%- if tools -%}
|
| 289 |
+
{{- "
|
| 290 |
+
Calls to these tools must go to the commentary channel: 'functions'." }}
|
| 291 |
+
{%- endif -%}
|
| 292 |
+
{%- endmacro -%}
|
| 293 |
+
|
| 294 |
+
{#- Main Template Logic ================================================= #}
|
| 295 |
+
{#- Set defaults #}
|
| 296 |
+
|
| 297 |
+
{#- Render system message #}
|
| 298 |
+
{{- "<|start|>system<|message|>" }}
|
| 299 |
+
{{- build_system_message() }}
|
| 300 |
+
{{- "<|end|>" }}
|
| 301 |
+
|
| 302 |
+
{#- Extract developer message #}
|
| 303 |
+
{%- if messages[0].role == "developer" or messages[0].role == "system" %}
|
| 304 |
+
{%- set developer_message = messages[0].content %}
|
| 305 |
+
{%- set loop_messages = messages[1:] %}
|
| 306 |
+
{%- else %}
|
| 307 |
+
{%- set developer_message = "" %}
|
| 308 |
+
{%- set loop_messages = messages %}
|
| 309 |
+
{%- endif %}
|
| 310 |
+
|
| 311 |
+
{#- Render developer message #}
|
| 312 |
+
{%- if developer_message or tools %}
|
| 313 |
+
{{- "<|start|>developer<|message|>" }}
|
| 314 |
+
{%- if developer_message %}
|
| 315 |
+
{{- "# Instructions
|
| 316 |
+
|
| 317 |
+
" }}
|
| 318 |
+
{{- developer_message }}
|
| 319 |
+
{%- endif %}
|
| 320 |
+
{%- if tools -%}
|
| 321 |
+
{{- "
|
| 322 |
+
|
| 323 |
+
" }}
|
| 324 |
+
{{- "# Tools
|
| 325 |
+
|
| 326 |
+
" }}
|
| 327 |
+
{{- render_tool_namespace("functions", tools) }}
|
| 328 |
+
{%- endif -%}
|
| 329 |
+
{{- "<|end|>" }}
|
| 330 |
+
{%- endif %}
|
| 331 |
+
|
| 332 |
+
{#- Render messages #}
|
| 333 |
+
{%- set last_tool_call = namespace(name=none) %}
|
| 334 |
+
{%- for message in loop_messages -%}
|
| 335 |
+
{#- At this point only assistant/user/tool messages should remain #}
|
| 336 |
+
{%- if message.role == 'assistant' -%}
|
| 337 |
+
{#- Checks to ensure the messages are being passed in the format we expect #}
|
| 338 |
+
{%- if "content" in message %}
|
| 339 |
+
{%- if "<|channel|>analysis<|message|>" in message.content or "<|channel|>final<|message|>" in message.content %}
|
| 340 |
+
{{- raise_exception("You have passed a message containing <|channel|> tags in the content field. Instead of doing this, you should pass analysis messages (the string between '<|message|>' and '<|end|>') in the 'thinking' field, and final messages (the string between '<|message|>' and '<|end|>') in the 'content' field.") }}
|
| 341 |
+
{%- endif %}
|
| 342 |
+
{%- endif %}
|
| 343 |
+
{%- if "thinking" in message %}
|
| 344 |
+
{%- if "<|channel|>analysis<|message|>" in message.thinking or "<|channel|>final<|message|>" in message.thinking %}
|
| 345 |
+
{{- raise_exception("You have passed a message containing <|channel|> tags in the thinking field. Instead of doing this, you should pass analysis messages (the string between '<|message|>' and '<|end|>') in the 'thinking' field, and final messages (the string between '<|message|>' and '<|end|>') in the 'content' field.") }}
|
| 346 |
+
{%- endif %}
|
| 347 |
+
{%- endif %}
|
| 348 |
+
{%- if "tool_calls" in message %}
|
| 349 |
+
{#- We assume max 1 tool call per message, and so we infer the tool call name #}
|
| 350 |
+
{#- in "tool" messages from the most recent assistant tool call name #}
|
| 351 |
+
{%- set tool_call = message.tool_calls[0] %}
|
| 352 |
+
{%- if tool_call.function %}
|
| 353 |
+
{%- set tool_call = tool_call.function %}
|
| 354 |
+
{%- endif %}
|
| 355 |
+
{%- if message.content and message.thinking %}
|
| 356 |
+
{{- raise_exception("Cannot pass both content and thinking in an assistant message with tool calls! Put the analysis message in one or the other, but not both.") }}
|
| 357 |
+
{%- elif message.content %}
|
| 358 |
+
{{- "<|start|>assistant<|channel|>analysis<|message|>" + message.content + "<|end|>" }}
|
| 359 |
+
{%- elif message.thinking %}
|
| 360 |
+
{{- "<|start|>assistant<|channel|>analysis<|message|>" + message.thinking + "<|end|>" }}
|
| 361 |
+
{%- endif %}
|
| 362 |
+
{{- "<|start|>assistant to=" }}
|
| 363 |
+
{{- "functions." + tool_call.name + "<|channel|>commentary " }}
|
| 364 |
+
{{- (tool_call.content_type if tool_call.content_type is defined else "json") + "<|message|>" }}
|
| 365 |
+
{{- tool_call.arguments|tojson }}
|
| 366 |
+
{{- "<|call|>" }}
|
| 367 |
+
{%- set last_tool_call.name = tool_call.name %}
|
| 368 |
+
{%- elif loop.last and not add_generation_prompt %}
|
| 369 |
+
{#- Only render the CoT if the final turn is an assistant turn and add_generation_prompt is false #}
|
| 370 |
+
{#- This is a situation that should only occur in training, never in inference. #}
|
| 371 |
+
{%- if "thinking" in message %}
|
| 372 |
+
{{- "<|start|>assistant<|channel|>analysis<|message|>" + message.thinking + "<|end|>" }}
|
| 373 |
+
{%- endif %}
|
| 374 |
+
{#- <|return|> indicates the end of generation, but <|end|> does not #}
|
| 375 |
+
{#- <|return|> should never be an input to the model, but we include it as the final token #}
|
| 376 |
+
{#- when training, so the model learns to emit it. #}
|
| 377 |
+
{{- "<|start|>assistant<|channel|>final<|message|>" + message.content + "<|return|>" }}
|
| 378 |
+
{%- else %}
|
| 379 |
+
{#- CoT is dropped during all previous turns, so we never render it for inference #}
|
| 380 |
+
{{- "<|start|>assistant<|channel|>final<|message|>" + message.content + "<|end|>" }}
|
| 381 |
+
{%- set last_tool_call.name = none %}
|
| 382 |
+
{%- endif %}
|
| 383 |
+
{%- elif message.role == 'tool' -%}
|
| 384 |
+
{%- if last_tool_call.name is none %}
|
| 385 |
+
{{- raise_exception("Message has tool role, but there was no previous assistant message with a tool call!") }}
|
| 386 |
+
{%- endif %}
|
| 387 |
+
{{- "<|start|>functions." + last_tool_call.name }}
|
| 388 |
+
{{- " to=assistant<|channel|>commentary<|message|>" + message.content|tojson + "<|end|>" }}
|
| 389 |
+
{%- elif message.role == 'user' -%}
|
| 390 |
+
{{- "<|start|>user<|message|>" + message.content + "<|end|>" }}
|
| 391 |
+
{%- endif -%}
|
| 392 |
+
{%- endfor -%}
|
| 393 |
+
|
| 394 |
+
{#- Generation prompt #}
|
| 395 |
+
{%- if add_generation_prompt -%}
|
| 396 |
+
<|start|>assistant
|
| 397 |
+
{%- endif -%}
|
config.json
ADDED
|
@@ -0,0 +1,76 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"GptOssForCausalLM"
|
| 4 |
+
],
|
| 5 |
+
"attention_bias": true,
|
| 6 |
+
"attention_dropout": 0.0,
|
| 7 |
+
"eos_token_id": 200002,
|
| 8 |
+
"experts_per_token": 4,
|
| 9 |
+
"head_dim": 64,
|
| 10 |
+
"hidden_act": "silu",
|
| 11 |
+
"hidden_size": 2880,
|
| 12 |
+
"initial_context_length": 4096,
|
| 13 |
+
"initializer_range": 0.02,
|
| 14 |
+
"intermediate_size": 2880,
|
| 15 |
+
"layer_types": [
|
| 16 |
+
"sliding_attention",
|
| 17 |
+
"full_attention",
|
| 18 |
+
"sliding_attention",
|
| 19 |
+
"full_attention",
|
| 20 |
+
"sliding_attention",
|
| 21 |
+
"full_attention",
|
| 22 |
+
"sliding_attention",
|
| 23 |
+
"full_attention",
|
| 24 |
+
"sliding_attention",
|
| 25 |
+
"full_attention",
|
| 26 |
+
"sliding_attention",
|
| 27 |
+
"full_attention",
|
| 28 |
+
"sliding_attention",
|
| 29 |
+
"full_attention",
|
| 30 |
+
"sliding_attention",
|
| 31 |
+
"full_attention",
|
| 32 |
+
"sliding_attention",
|
| 33 |
+
"full_attention",
|
| 34 |
+
"sliding_attention",
|
| 35 |
+
"full_attention",
|
| 36 |
+
"sliding_attention",
|
| 37 |
+
"full_attention",
|
| 38 |
+
"sliding_attention",
|
| 39 |
+
"full_attention"
|
| 40 |
+
],
|
| 41 |
+
"max_position_embeddings": 131072,
|
| 42 |
+
"model_type": "gpt_oss",
|
| 43 |
+
"num_attention_heads": 64,
|
| 44 |
+
"num_experts_per_tok": 4,
|
| 45 |
+
"num_hidden_layers": 24,
|
| 46 |
+
"num_key_value_heads": 8,
|
| 47 |
+
"num_local_experts": 32,
|
| 48 |
+
"output_router_logits": false,
|
| 49 |
+
"pad_token_id": 199999,
|
| 50 |
+
"quantization_config": {
|
| 51 |
+
"modules_to_not_convert": [
|
| 52 |
+
"model.layers.*.self_attn",
|
| 53 |
+
"model.layers.*.mlp.router",
|
| 54 |
+
"model.embed_tokens",
|
| 55 |
+
"lm_head"
|
| 56 |
+
],
|
| 57 |
+
"quant_method": "mxfp4"
|
| 58 |
+
},
|
| 59 |
+
"rms_norm_eps": 1e-05,
|
| 60 |
+
"rope_scaling": {
|
| 61 |
+
"beta_fast": 32.0,
|
| 62 |
+
"beta_slow": 1.0,
|
| 63 |
+
"factor": 32.0,
|
| 64 |
+
"original_max_position_embeddings": 4096,
|
| 65 |
+
"rope_type": "yarn",
|
| 66 |
+
"truncate": false
|
| 67 |
+
},
|
| 68 |
+
"rope_theta": 150000,
|
| 69 |
+
"router_aux_loss_coef": 0.9,
|
| 70 |
+
"sliding_window": 128,
|
| 71 |
+
"swiglu_limit": 7.0,
|
| 72 |
+
"tie_word_embeddings": false,
|
| 73 |
+
"transformers_version": "4.55.0.dev0",
|
| 74 |
+
"use_cache": true,
|
| 75 |
+
"vocab_size": 201088
|
| 76 |
+
}
|
generation_config.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"bos_token_id": 199998,
|
| 3 |
+
"do_sample": true,
|
| 4 |
+
"eos_token_id": [
|
| 5 |
+
200002,
|
| 6 |
+
199999
|
| 7 |
+
],
|
| 8 |
+
"pad_token_id": 199999,
|
| 9 |
+
"transformers_version": "4.55.0.dev0"
|
| 10 |
+
}
|
inference.py
ADDED
|
@@ -0,0 +1,183 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/usr/bin/env python3
|
| 2 |
+
"""
|
| 3 |
+
Example inference script for CAI-20B Marketing Strategy Expert
|
| 4 |
+
"""
|
| 5 |
+
|
| 6 |
+
import torch
|
| 7 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 8 |
+
import re
|
| 9 |
+
|
| 10 |
+
class ResponseCleaner:
|
| 11 |
+
"""Clean up model responses to remove artifacts"""
|
| 12 |
+
|
| 13 |
+
def __init__(self):
|
| 14 |
+
self.artifact_patterns = [
|
| 15 |
+
r'<\|[^>]+\|>', # Special tokens
|
| 16 |
+
r'assistantfinal',
|
| 17 |
+
r'assistant\s*final',
|
| 18 |
+
r'We need to.*?(?=\n|$)',
|
| 19 |
+
r'Let me.*?(?=\n|$)',
|
| 20 |
+
r'I need to.*?(?=\n|$)',
|
| 21 |
+
r'\\n\\n\\n+', # Multiple newlines
|
| 22 |
+
]
|
| 23 |
+
|
| 24 |
+
def clean_response(self, text):
|
| 25 |
+
"""Clean artifacts from response"""
|
| 26 |
+
cleaned = text
|
| 27 |
+
|
| 28 |
+
# Remove artifacts
|
| 29 |
+
for pattern in self.artifact_patterns:
|
| 30 |
+
cleaned = re.sub(pattern, '', cleaned, flags=re.IGNORECASE)
|
| 31 |
+
|
| 32 |
+
# Clean up spacing
|
| 33 |
+
cleaned = re.sub(r'\s+', ' ', cleaned).strip()
|
| 34 |
+
|
| 35 |
+
# Ensure proper ending
|
| 36 |
+
if cleaned and cleaned[-1] not in '.!?':
|
| 37 |
+
last_sentence = cleaned.split('.')[-1].strip()
|
| 38 |
+
if len(last_sentence) < 20:
|
| 39 |
+
parts = cleaned.rsplit('.', 1)
|
| 40 |
+
if len(parts) > 1:
|
| 41 |
+
cleaned = parts[0] + '.'
|
| 42 |
+
else:
|
| 43 |
+
cleaned += '.'
|
| 44 |
+
|
| 45 |
+
return cleaned
|
| 46 |
+
|
| 47 |
+
|
| 48 |
+
class CAI20BMarketing:
|
| 49 |
+
"""CAI-20B Marketing Strategy Expert Model"""
|
| 50 |
+
|
| 51 |
+
def __init__(self, model_name="tigres2526/CAI-20B", device="cuda"):
|
| 52 |
+
"""Initialize the model and tokenizer"""
|
| 53 |
+
print("Loading CAI-20B Marketing Strategy Expert...")
|
| 54 |
+
|
| 55 |
+
self.device = device
|
| 56 |
+
self.cleaner = ResponseCleaner()
|
| 57 |
+
|
| 58 |
+
# Load tokenizer
|
| 59 |
+
self.tokenizer = AutoTokenizer.from_pretrained(
|
| 60 |
+
model_name,
|
| 61 |
+
trust_remote_code=True
|
| 62 |
+
)
|
| 63 |
+
self.tokenizer.pad_token = self.tokenizer.eos_token
|
| 64 |
+
|
| 65 |
+
# Load model
|
| 66 |
+
self.model = AutoModelForCausalLM.from_pretrained(
|
| 67 |
+
model_name,
|
| 68 |
+
device_map="auto",
|
| 69 |
+
torch_dtype=torch.bfloat16,
|
| 70 |
+
trust_remote_code=True
|
| 71 |
+
)
|
| 72 |
+
self.model.eval()
|
| 73 |
+
|
| 74 |
+
print("✅ Model loaded successfully!")
|
| 75 |
+
|
| 76 |
+
def generate(
|
| 77 |
+
self,
|
| 78 |
+
question,
|
| 79 |
+
max_new_tokens=250,
|
| 80 |
+
temperature=0.7,
|
| 81 |
+
top_p=0.9,
|
| 82 |
+
repetition_penalty=1.1,
|
| 83 |
+
clean_output=True
|
| 84 |
+
):
|
| 85 |
+
"""Generate marketing advice for a given question"""
|
| 86 |
+
|
| 87 |
+
# Format prompt
|
| 88 |
+
prompt = f"""You are a marketing strategy expert specializing in performance marketing, creative development, and conversion optimization.
|
| 89 |
+
Provide practical, actionable advice grounded in real-world experience.
|
| 90 |
+
|
| 91 |
+
User: {question}
|
| 92 |
+
Assistant:"""
|
| 93 |
+
|
| 94 |
+
# Tokenize
|
| 95 |
+
inputs = self.tokenizer(
|
| 96 |
+
prompt,
|
| 97 |
+
return_tensors="pt",
|
| 98 |
+
truncation=True,
|
| 99 |
+
max_length=2048
|
| 100 |
+
).to(self.model.device)
|
| 101 |
+
|
| 102 |
+
# Generate
|
| 103 |
+
with torch.no_grad():
|
| 104 |
+
outputs = self.model.generate(
|
| 105 |
+
**inputs,
|
| 106 |
+
max_new_tokens=max_new_tokens,
|
| 107 |
+
temperature=temperature,
|
| 108 |
+
top_p=top_p,
|
| 109 |
+
repetition_penalty=repetition_penalty,
|
| 110 |
+
do_sample=True,
|
| 111 |
+
pad_token_id=self.tokenizer.pad_token_id,
|
| 112 |
+
eos_token_id=self.tokenizer.eos_token_id,
|
| 113 |
+
no_repeat_ngram_size=3,
|
| 114 |
+
)
|
| 115 |
+
|
| 116 |
+
# Decode
|
| 117 |
+
response = self.tokenizer.decode(
|
| 118 |
+
outputs[0][inputs['input_ids'].shape[1]:],
|
| 119 |
+
skip_special_tokens=True
|
| 120 |
+
)
|
| 121 |
+
|
| 122 |
+
# Clean if requested
|
| 123 |
+
if clean_output:
|
| 124 |
+
response = self.cleaner.clean_response(response)
|
| 125 |
+
|
| 126 |
+
return response
|
| 127 |
+
|
| 128 |
+
def chat(self):
|
| 129 |
+
"""Interactive chat mode"""
|
| 130 |
+
print("\n" + "=" * 70)
|
| 131 |
+
print("CAI-20B Marketing Strategy Expert - Interactive Chat")
|
| 132 |
+
print("Type 'exit' to quit")
|
| 133 |
+
print("=" * 70 + "\n")
|
| 134 |
+
|
| 135 |
+
while True:
|
| 136 |
+
user_input = input("You: ").strip()
|
| 137 |
+
|
| 138 |
+
if user_input.lower() == 'exit':
|
| 139 |
+
print("Goodbye!")
|
| 140 |
+
break
|
| 141 |
+
|
| 142 |
+
if not user_input:
|
| 143 |
+
continue
|
| 144 |
+
|
| 145 |
+
# Generate response
|
| 146 |
+
response = self.generate(user_input)
|
| 147 |
+
|
| 148 |
+
# Display
|
| 149 |
+
print(f"\nCAI-20B: {response}\n")
|
| 150 |
+
print("-" * 70 + "\n")
|
| 151 |
+
|
| 152 |
+
|
| 153 |
+
def main():
|
| 154 |
+
"""Example usage"""
|
| 155 |
+
|
| 156 |
+
# Initialize model
|
| 157 |
+
model = CAI20BMarketing()
|
| 158 |
+
|
| 159 |
+
# Example questions
|
| 160 |
+
test_questions = [
|
| 161 |
+
"What are the top 3 marketing channels for a B2B SaaS startup?",
|
| 162 |
+
"How should I allocate a $10K monthly marketing budget?",
|
| 163 |
+
"What's the difference between CAC and LTV?",
|
| 164 |
+
]
|
| 165 |
+
|
| 166 |
+
print("\n" + "=" * 70)
|
| 167 |
+
print("Running example questions...")
|
| 168 |
+
print("=" * 70 + "\n")
|
| 169 |
+
|
| 170 |
+
for i, question in enumerate(test_questions, 1):
|
| 171 |
+
print(f"Q{i}: {question}")
|
| 172 |
+
response = model.generate(question)
|
| 173 |
+
print(f"A: {response}\n")
|
| 174 |
+
print("-" * 50 + "\n")
|
| 175 |
+
|
| 176 |
+
# Optional: Start interactive chat
|
| 177 |
+
print("\nWould you like to start interactive chat? (y/n)")
|
| 178 |
+
if input().lower() == 'y':
|
| 179 |
+
model.chat()
|
| 180 |
+
|
| 181 |
+
|
| 182 |
+
if __name__ == "__main__":
|
| 183 |
+
main()
|
model-00001-of-00009.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:616382eda8c6eeeb1be6a7fa8aa5a7eccc2bcf2444545343b5d26659f42d9130
|
| 3 |
+
size 4504304664
|
model-00002-of-00009.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:787011c75a4bbb47d107ca304c55e0d8e4ebf5e252330d0829c04f914e1c04b5
|
| 3 |
+
size 4939127656
|
model-00003-of-00009.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d572c644051a0ba941e6fb2fc256d47a52aa3c939e83325dd2b50c1806b45477
|
| 3 |
+
size 4939127656
|
model-00004-of-00009.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c0c31fe726ddf2073bedff4ab029920785631b508ada90d75f5306e5dcde4c30
|
| 3 |
+
size 4939127680
|
model-00005-of-00009.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:00e082e4cdc85c872f135e1035724ccab60185f53d2dd54c381c6d156afa49a5
|
| 3 |
+
size 4939127704
|
model-00006-of-00009.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9fca004bd72be8cf7ff8de26135124e7512cd2fb4470b8bc8a25efbbe0d3bdf5
|
| 3 |
+
size 4939127704
|
model-00007-of-00009.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:99484514a77005463c199038aff230542a49075ae64b292558ab86c88db204da
|
| 3 |
+
size 4939127704
|
model-00008-of-00009.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:59a3c5c53d5e613cec797baa079f2e2067c9719755d26c919c7240376d77f1bf
|
| 3 |
+
size 4939127704
|
model-00009-of-00009.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6c1951ff65d233bcb7f50485d294cced6497b8e0d9707f57087912fc2352492e
|
| 3 |
+
size 2751362856
|
model.safetensors.index.json
ADDED
|
@@ -0,0 +1,419 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"metadata": {
|
| 3 |
+
"total_parameters": 20914757184,
|
| 4 |
+
"total_size": 41829514368
|
| 5 |
+
},
|
| 6 |
+
"weight_map": {
|
| 7 |
+
"lm_head.weight": "model-00009-of-00009.safetensors",
|
| 8 |
+
"model.embed_tokens.weight": "model-00001-of-00009.safetensors",
|
| 9 |
+
"model.layers.0.input_layernorm.weight": "model-00001-of-00009.safetensors",
|
| 10 |
+
"model.layers.0.mlp.experts.down_proj": "model-00001-of-00009.safetensors",
|
| 11 |
+
"model.layers.0.mlp.experts.down_proj_bias": "model-00001-of-00009.safetensors",
|
| 12 |
+
"model.layers.0.mlp.experts.gate_up_proj": "model-00001-of-00009.safetensors",
|
| 13 |
+
"model.layers.0.mlp.experts.gate_up_proj_bias": "model-00001-of-00009.safetensors",
|
| 14 |
+
"model.layers.0.mlp.router.bias": "model-00001-of-00009.safetensors",
|
| 15 |
+
"model.layers.0.mlp.router.weight": "model-00001-of-00009.safetensors",
|
| 16 |
+
"model.layers.0.post_attention_layernorm.weight": "model-00001-of-00009.safetensors",
|
| 17 |
+
"model.layers.0.self_attn.k_proj.bias": "model-00001-of-00009.safetensors",
|
| 18 |
+
"model.layers.0.self_attn.k_proj.weight": "model-00001-of-00009.safetensors",
|
| 19 |
+
"model.layers.0.self_attn.o_proj.bias": "model-00001-of-00009.safetensors",
|
| 20 |
+
"model.layers.0.self_attn.o_proj.weight": "model-00001-of-00009.safetensors",
|
| 21 |
+
"model.layers.0.self_attn.q_proj.bias": "model-00001-of-00009.safetensors",
|
| 22 |
+
"model.layers.0.self_attn.q_proj.weight": "model-00001-of-00009.safetensors",
|
| 23 |
+
"model.layers.0.self_attn.sinks": "model-00001-of-00009.safetensors",
|
| 24 |
+
"model.layers.0.self_attn.v_proj.bias": "model-00001-of-00009.safetensors",
|
| 25 |
+
"model.layers.0.self_attn.v_proj.weight": "model-00001-of-00009.safetensors",
|
| 26 |
+
"model.layers.1.input_layernorm.weight": "model-00001-of-00009.safetensors",
|
| 27 |
+
"model.layers.1.mlp.experts.down_proj": "model-00001-of-00009.safetensors",
|
| 28 |
+
"model.layers.1.mlp.experts.down_proj_bias": "model-00001-of-00009.safetensors",
|
| 29 |
+
"model.layers.1.mlp.experts.gate_up_proj": "model-00001-of-00009.safetensors",
|
| 30 |
+
"model.layers.1.mlp.experts.gate_up_proj_bias": "model-00001-of-00009.safetensors",
|
| 31 |
+
"model.layers.1.mlp.router.bias": "model-00001-of-00009.safetensors",
|
| 32 |
+
"model.layers.1.mlp.router.weight": "model-00001-of-00009.safetensors",
|
| 33 |
+
"model.layers.1.post_attention_layernorm.weight": "model-00001-of-00009.safetensors",
|
| 34 |
+
"model.layers.1.self_attn.k_proj.bias": "model-00001-of-00009.safetensors",
|
| 35 |
+
"model.layers.1.self_attn.k_proj.weight": "model-00001-of-00009.safetensors",
|
| 36 |
+
"model.layers.1.self_attn.o_proj.bias": "model-00001-of-00009.safetensors",
|
| 37 |
+
"model.layers.1.self_attn.o_proj.weight": "model-00001-of-00009.safetensors",
|
| 38 |
+
"model.layers.1.self_attn.q_proj.bias": "model-00001-of-00009.safetensors",
|
| 39 |
+
"model.layers.1.self_attn.q_proj.weight": "model-00001-of-00009.safetensors",
|
| 40 |
+
"model.layers.1.self_attn.sinks": "model-00001-of-00009.safetensors",
|
| 41 |
+
"model.layers.1.self_attn.v_proj.bias": "model-00001-of-00009.safetensors",
|
| 42 |
+
"model.layers.1.self_attn.v_proj.weight": "model-00001-of-00009.safetensors",
|
| 43 |
+
"model.layers.10.input_layernorm.weight": "model-00004-of-00009.safetensors",
|
| 44 |
+
"model.layers.10.mlp.experts.down_proj": "model-00004-of-00009.safetensors",
|
| 45 |
+
"model.layers.10.mlp.experts.down_proj_bias": "model-00004-of-00009.safetensors",
|
| 46 |
+
"model.layers.10.mlp.experts.gate_up_proj": "model-00004-of-00009.safetensors",
|
| 47 |
+
"model.layers.10.mlp.experts.gate_up_proj_bias": "model-00004-of-00009.safetensors",
|
| 48 |
+
"model.layers.10.mlp.router.bias": "model-00004-of-00009.safetensors",
|
| 49 |
+
"model.layers.10.mlp.router.weight": "model-00004-of-00009.safetensors",
|
| 50 |
+
"model.layers.10.post_attention_layernorm.weight": "model-00004-of-00009.safetensors",
|
| 51 |
+
"model.layers.10.self_attn.k_proj.bias": "model-00004-of-00009.safetensors",
|
| 52 |
+
"model.layers.10.self_attn.k_proj.weight": "model-00004-of-00009.safetensors",
|
| 53 |
+
"model.layers.10.self_attn.o_proj.bias": "model-00004-of-00009.safetensors",
|
| 54 |
+
"model.layers.10.self_attn.o_proj.weight": "model-00004-of-00009.safetensors",
|
| 55 |
+
"model.layers.10.self_attn.q_proj.bias": "model-00004-of-00009.safetensors",
|
| 56 |
+
"model.layers.10.self_attn.q_proj.weight": "model-00004-of-00009.safetensors",
|
| 57 |
+
"model.layers.10.self_attn.sinks": "model-00004-of-00009.safetensors",
|
| 58 |
+
"model.layers.10.self_attn.v_proj.bias": "model-00004-of-00009.safetensors",
|
| 59 |
+
"model.layers.10.self_attn.v_proj.weight": "model-00004-of-00009.safetensors",
|
| 60 |
+
"model.layers.11.input_layernorm.weight": "model-00005-of-00009.safetensors",
|
| 61 |
+
"model.layers.11.mlp.experts.down_proj": "model-00005-of-00009.safetensors",
|
| 62 |
+
"model.layers.11.mlp.experts.down_proj_bias": "model-00005-of-00009.safetensors",
|
| 63 |
+
"model.layers.11.mlp.experts.gate_up_proj": "model-00005-of-00009.safetensors",
|
| 64 |
+
"model.layers.11.mlp.experts.gate_up_proj_bias": "model-00005-of-00009.safetensors",
|
| 65 |
+
"model.layers.11.mlp.router.bias": "model-00004-of-00009.safetensors",
|
| 66 |
+
"model.layers.11.mlp.router.weight": "model-00004-of-00009.safetensors",
|
| 67 |
+
"model.layers.11.post_attention_layernorm.weight": "model-00005-of-00009.safetensors",
|
| 68 |
+
"model.layers.11.self_attn.k_proj.bias": "model-00004-of-00009.safetensors",
|
| 69 |
+
"model.layers.11.self_attn.k_proj.weight": "model-00004-of-00009.safetensors",
|
| 70 |
+
"model.layers.11.self_attn.o_proj.bias": "model-00004-of-00009.safetensors",
|
| 71 |
+
"model.layers.11.self_attn.o_proj.weight": "model-00004-of-00009.safetensors",
|
| 72 |
+
"model.layers.11.self_attn.q_proj.bias": "model-00004-of-00009.safetensors",
|
| 73 |
+
"model.layers.11.self_attn.q_proj.weight": "model-00004-of-00009.safetensors",
|
| 74 |
+
"model.layers.11.self_attn.sinks": "model-00004-of-00009.safetensors",
|
| 75 |
+
"model.layers.11.self_attn.v_proj.bias": "model-00004-of-00009.safetensors",
|
| 76 |
+
"model.layers.11.self_attn.v_proj.weight": "model-00004-of-00009.safetensors",
|
| 77 |
+
"model.layers.12.input_layernorm.weight": "model-00005-of-00009.safetensors",
|
| 78 |
+
"model.layers.12.mlp.experts.down_proj": "model-00005-of-00009.safetensors",
|
| 79 |
+
"model.layers.12.mlp.experts.down_proj_bias": "model-00005-of-00009.safetensors",
|
| 80 |
+
"model.layers.12.mlp.experts.gate_up_proj": "model-00005-of-00009.safetensors",
|
| 81 |
+
"model.layers.12.mlp.experts.gate_up_proj_bias": "model-00005-of-00009.safetensors",
|
| 82 |
+
"model.layers.12.mlp.router.bias": "model-00005-of-00009.safetensors",
|
| 83 |
+
"model.layers.12.mlp.router.weight": "model-00005-of-00009.safetensors",
|
| 84 |
+
"model.layers.12.post_attention_layernorm.weight": "model-00005-of-00009.safetensors",
|
| 85 |
+
"model.layers.12.self_attn.k_proj.bias": "model-00005-of-00009.safetensors",
|
| 86 |
+
"model.layers.12.self_attn.k_proj.weight": "model-00005-of-00009.safetensors",
|
| 87 |
+
"model.layers.12.self_attn.o_proj.bias": "model-00005-of-00009.safetensors",
|
| 88 |
+
"model.layers.12.self_attn.o_proj.weight": "model-00005-of-00009.safetensors",
|
| 89 |
+
"model.layers.12.self_attn.q_proj.bias": "model-00005-of-00009.safetensors",
|
| 90 |
+
"model.layers.12.self_attn.q_proj.weight": "model-00005-of-00009.safetensors",
|
| 91 |
+
"model.layers.12.self_attn.sinks": "model-00005-of-00009.safetensors",
|
| 92 |
+
"model.layers.12.self_attn.v_proj.bias": "model-00005-of-00009.safetensors",
|
| 93 |
+
"model.layers.12.self_attn.v_proj.weight": "model-00005-of-00009.safetensors",
|
| 94 |
+
"model.layers.13.input_layernorm.weight": "model-00005-of-00009.safetensors",
|
| 95 |
+
"model.layers.13.mlp.experts.down_proj": "model-00005-of-00009.safetensors",
|
| 96 |
+
"model.layers.13.mlp.experts.down_proj_bias": "model-00005-of-00009.safetensors",
|
| 97 |
+
"model.layers.13.mlp.experts.gate_up_proj": "model-00005-of-00009.safetensors",
|
| 98 |
+
"model.layers.13.mlp.experts.gate_up_proj_bias": "model-00005-of-00009.safetensors",
|
| 99 |
+
"model.layers.13.mlp.router.bias": "model-00005-of-00009.safetensors",
|
| 100 |
+
"model.layers.13.mlp.router.weight": "model-00005-of-00009.safetensors",
|
| 101 |
+
"model.layers.13.post_attention_layernorm.weight": "model-00005-of-00009.safetensors",
|
| 102 |
+
"model.layers.13.self_attn.k_proj.bias": "model-00005-of-00009.safetensors",
|
| 103 |
+
"model.layers.13.self_attn.k_proj.weight": "model-00005-of-00009.safetensors",
|
| 104 |
+
"model.layers.13.self_attn.o_proj.bias": "model-00005-of-00009.safetensors",
|
| 105 |
+
"model.layers.13.self_attn.o_proj.weight": "model-00005-of-00009.safetensors",
|
| 106 |
+
"model.layers.13.self_attn.q_proj.bias": "model-00005-of-00009.safetensors",
|
| 107 |
+
"model.layers.13.self_attn.q_proj.weight": "model-00005-of-00009.safetensors",
|
| 108 |
+
"model.layers.13.self_attn.sinks": "model-00005-of-00009.safetensors",
|
| 109 |
+
"model.layers.13.self_attn.v_proj.bias": "model-00005-of-00009.safetensors",
|
| 110 |
+
"model.layers.13.self_attn.v_proj.weight": "model-00005-of-00009.safetensors",
|
| 111 |
+
"model.layers.14.input_layernorm.weight": "model-00006-of-00009.safetensors",
|
| 112 |
+
"model.layers.14.mlp.experts.down_proj": "model-00006-of-00009.safetensors",
|
| 113 |
+
"model.layers.14.mlp.experts.down_proj_bias": "model-00006-of-00009.safetensors",
|
| 114 |
+
"model.layers.14.mlp.experts.gate_up_proj": "model-00006-of-00009.safetensors",
|
| 115 |
+
"model.layers.14.mlp.experts.gate_up_proj_bias": "model-00006-of-00009.safetensors",
|
| 116 |
+
"model.layers.14.mlp.router.bias": "model-00005-of-00009.safetensors",
|
| 117 |
+
"model.layers.14.mlp.router.weight": "model-00005-of-00009.safetensors",
|
| 118 |
+
"model.layers.14.post_attention_layernorm.weight": "model-00006-of-00009.safetensors",
|
| 119 |
+
"model.layers.14.self_attn.k_proj.bias": "model-00005-of-00009.safetensors",
|
| 120 |
+
"model.layers.14.self_attn.k_proj.weight": "model-00005-of-00009.safetensors",
|
| 121 |
+
"model.layers.14.self_attn.o_proj.bias": "model-00005-of-00009.safetensors",
|
| 122 |
+
"model.layers.14.self_attn.o_proj.weight": "model-00005-of-00009.safetensors",
|
| 123 |
+
"model.layers.14.self_attn.q_proj.bias": "model-00005-of-00009.safetensors",
|
| 124 |
+
"model.layers.14.self_attn.q_proj.weight": "model-00005-of-00009.safetensors",
|
| 125 |
+
"model.layers.14.self_attn.sinks": "model-00005-of-00009.safetensors",
|
| 126 |
+
"model.layers.14.self_attn.v_proj.bias": "model-00005-of-00009.safetensors",
|
| 127 |
+
"model.layers.14.self_attn.v_proj.weight": "model-00005-of-00009.safetensors",
|
| 128 |
+
"model.layers.15.input_layernorm.weight": "model-00006-of-00009.safetensors",
|
| 129 |
+
"model.layers.15.mlp.experts.down_proj": "model-00006-of-00009.safetensors",
|
| 130 |
+
"model.layers.15.mlp.experts.down_proj_bias": "model-00006-of-00009.safetensors",
|
| 131 |
+
"model.layers.15.mlp.experts.gate_up_proj": "model-00006-of-00009.safetensors",
|
| 132 |
+
"model.layers.15.mlp.experts.gate_up_proj_bias": "model-00006-of-00009.safetensors",
|
| 133 |
+
"model.layers.15.mlp.router.bias": "model-00006-of-00009.safetensors",
|
| 134 |
+
"model.layers.15.mlp.router.weight": "model-00006-of-00009.safetensors",
|
| 135 |
+
"model.layers.15.post_attention_layernorm.weight": "model-00006-of-00009.safetensors",
|
| 136 |
+
"model.layers.15.self_attn.k_proj.bias": "model-00006-of-00009.safetensors",
|
| 137 |
+
"model.layers.15.self_attn.k_proj.weight": "model-00006-of-00009.safetensors",
|
| 138 |
+
"model.layers.15.self_attn.o_proj.bias": "model-00006-of-00009.safetensors",
|
| 139 |
+
"model.layers.15.self_attn.o_proj.weight": "model-00006-of-00009.safetensors",
|
| 140 |
+
"model.layers.15.self_attn.q_proj.bias": "model-00006-of-00009.safetensors",
|
| 141 |
+
"model.layers.15.self_attn.q_proj.weight": "model-00006-of-00009.safetensors",
|
| 142 |
+
"model.layers.15.self_attn.sinks": "model-00006-of-00009.safetensors",
|
| 143 |
+
"model.layers.15.self_attn.v_proj.bias": "model-00006-of-00009.safetensors",
|
| 144 |
+
"model.layers.15.self_attn.v_proj.weight": "model-00006-of-00009.safetensors",
|
| 145 |
+
"model.layers.16.input_layernorm.weight": "model-00006-of-00009.safetensors",
|
| 146 |
+
"model.layers.16.mlp.experts.down_proj": "model-00006-of-00009.safetensors",
|
| 147 |
+
"model.layers.16.mlp.experts.down_proj_bias": "model-00006-of-00009.safetensors",
|
| 148 |
+
"model.layers.16.mlp.experts.gate_up_proj": "model-00006-of-00009.safetensors",
|
| 149 |
+
"model.layers.16.mlp.experts.gate_up_proj_bias": "model-00006-of-00009.safetensors",
|
| 150 |
+
"model.layers.16.mlp.router.bias": "model-00006-of-00009.safetensors",
|
| 151 |
+
"model.layers.16.mlp.router.weight": "model-00006-of-00009.safetensors",
|
| 152 |
+
"model.layers.16.post_attention_layernorm.weight": "model-00006-of-00009.safetensors",
|
| 153 |
+
"model.layers.16.self_attn.k_proj.bias": "model-00006-of-00009.safetensors",
|
| 154 |
+
"model.layers.16.self_attn.k_proj.weight": "model-00006-of-00009.safetensors",
|
| 155 |
+
"model.layers.16.self_attn.o_proj.bias": "model-00006-of-00009.safetensors",
|
| 156 |
+
"model.layers.16.self_attn.o_proj.weight": "model-00006-of-00009.safetensors",
|
| 157 |
+
"model.layers.16.self_attn.q_proj.bias": "model-00006-of-00009.safetensors",
|
| 158 |
+
"model.layers.16.self_attn.q_proj.weight": "model-00006-of-00009.safetensors",
|
| 159 |
+
"model.layers.16.self_attn.sinks": "model-00006-of-00009.safetensors",
|
| 160 |
+
"model.layers.16.self_attn.v_proj.bias": "model-00006-of-00009.safetensors",
|
| 161 |
+
"model.layers.16.self_attn.v_proj.weight": "model-00006-of-00009.safetensors",
|
| 162 |
+
"model.layers.17.input_layernorm.weight": "model-00007-of-00009.safetensors",
|
| 163 |
+
"model.layers.17.mlp.experts.down_proj": "model-00007-of-00009.safetensors",
|
| 164 |
+
"model.layers.17.mlp.experts.down_proj_bias": "model-00007-of-00009.safetensors",
|
| 165 |
+
"model.layers.17.mlp.experts.gate_up_proj": "model-00007-of-00009.safetensors",
|
| 166 |
+
"model.layers.17.mlp.experts.gate_up_proj_bias": "model-00007-of-00009.safetensors",
|
| 167 |
+
"model.layers.17.mlp.router.bias": "model-00006-of-00009.safetensors",
|
| 168 |
+
"model.layers.17.mlp.router.weight": "model-00006-of-00009.safetensors",
|
| 169 |
+
"model.layers.17.post_attention_layernorm.weight": "model-00007-of-00009.safetensors",
|
| 170 |
+
"model.layers.17.self_attn.k_proj.bias": "model-00006-of-00009.safetensors",
|
| 171 |
+
"model.layers.17.self_attn.k_proj.weight": "model-00006-of-00009.safetensors",
|
| 172 |
+
"model.layers.17.self_attn.o_proj.bias": "model-00006-of-00009.safetensors",
|
| 173 |
+
"model.layers.17.self_attn.o_proj.weight": "model-00006-of-00009.safetensors",
|
| 174 |
+
"model.layers.17.self_attn.q_proj.bias": "model-00006-of-00009.safetensors",
|
| 175 |
+
"model.layers.17.self_attn.q_proj.weight": "model-00006-of-00009.safetensors",
|
| 176 |
+
"model.layers.17.self_attn.sinks": "model-00006-of-00009.safetensors",
|
| 177 |
+
"model.layers.17.self_attn.v_proj.bias": "model-00006-of-00009.safetensors",
|
| 178 |
+
"model.layers.17.self_attn.v_proj.weight": "model-00006-of-00009.safetensors",
|
| 179 |
+
"model.layers.18.input_layernorm.weight": "model-00007-of-00009.safetensors",
|
| 180 |
+
"model.layers.18.mlp.experts.down_proj": "model-00007-of-00009.safetensors",
|
| 181 |
+
"model.layers.18.mlp.experts.down_proj_bias": "model-00007-of-00009.safetensors",
|
| 182 |
+
"model.layers.18.mlp.experts.gate_up_proj": "model-00007-of-00009.safetensors",
|
| 183 |
+
"model.layers.18.mlp.experts.gate_up_proj_bias": "model-00007-of-00009.safetensors",
|
| 184 |
+
"model.layers.18.mlp.router.bias": "model-00007-of-00009.safetensors",
|
| 185 |
+
"model.layers.18.mlp.router.weight": "model-00007-of-00009.safetensors",
|
| 186 |
+
"model.layers.18.post_attention_layernorm.weight": "model-00007-of-00009.safetensors",
|
| 187 |
+
"model.layers.18.self_attn.k_proj.bias": "model-00007-of-00009.safetensors",
|
| 188 |
+
"model.layers.18.self_attn.k_proj.weight": "model-00007-of-00009.safetensors",
|
| 189 |
+
"model.layers.18.self_attn.o_proj.bias": "model-00007-of-00009.safetensors",
|
| 190 |
+
"model.layers.18.self_attn.o_proj.weight": "model-00007-of-00009.safetensors",
|
| 191 |
+
"model.layers.18.self_attn.q_proj.bias": "model-00007-of-00009.safetensors",
|
| 192 |
+
"model.layers.18.self_attn.q_proj.weight": "model-00007-of-00009.safetensors",
|
| 193 |
+
"model.layers.18.self_attn.sinks": "model-00007-of-00009.safetensors",
|
| 194 |
+
"model.layers.18.self_attn.v_proj.bias": "model-00007-of-00009.safetensors",
|
| 195 |
+
"model.layers.18.self_attn.v_proj.weight": "model-00007-of-00009.safetensors",
|
| 196 |
+
"model.layers.19.input_layernorm.weight": "model-00007-of-00009.safetensors",
|
| 197 |
+
"model.layers.19.mlp.experts.down_proj": "model-00007-of-00009.safetensors",
|
| 198 |
+
"model.layers.19.mlp.experts.down_proj_bias": "model-00007-of-00009.safetensors",
|
| 199 |
+
"model.layers.19.mlp.experts.gate_up_proj": "model-00007-of-00009.safetensors",
|
| 200 |
+
"model.layers.19.mlp.experts.gate_up_proj_bias": "model-00007-of-00009.safetensors",
|
| 201 |
+
"model.layers.19.mlp.router.bias": "model-00007-of-00009.safetensors",
|
| 202 |
+
"model.layers.19.mlp.router.weight": "model-00007-of-00009.safetensors",
|
| 203 |
+
"model.layers.19.post_attention_layernorm.weight": "model-00007-of-00009.safetensors",
|
| 204 |
+
"model.layers.19.self_attn.k_proj.bias": "model-00007-of-00009.safetensors",
|
| 205 |
+
"model.layers.19.self_attn.k_proj.weight": "model-00007-of-00009.safetensors",
|
| 206 |
+
"model.layers.19.self_attn.o_proj.bias": "model-00007-of-00009.safetensors",
|
| 207 |
+
"model.layers.19.self_attn.o_proj.weight": "model-00007-of-00009.safetensors",
|
| 208 |
+
"model.layers.19.self_attn.q_proj.bias": "model-00007-of-00009.safetensors",
|
| 209 |
+
"model.layers.19.self_attn.q_proj.weight": "model-00007-of-00009.safetensors",
|
| 210 |
+
"model.layers.19.self_attn.sinks": "model-00007-of-00009.safetensors",
|
| 211 |
+
"model.layers.19.self_attn.v_proj.bias": "model-00007-of-00009.safetensors",
|
| 212 |
+
"model.layers.19.self_attn.v_proj.weight": "model-00007-of-00009.safetensors",
|
| 213 |
+
"model.layers.2.input_layernorm.weight": "model-00002-of-00009.safetensors",
|
| 214 |
+
"model.layers.2.mlp.experts.down_proj": "model-00002-of-00009.safetensors",
|
| 215 |
+
"model.layers.2.mlp.experts.down_proj_bias": "model-00002-of-00009.safetensors",
|
| 216 |
+
"model.layers.2.mlp.experts.gate_up_proj": "model-00002-of-00009.safetensors",
|
| 217 |
+
"model.layers.2.mlp.experts.gate_up_proj_bias": "model-00002-of-00009.safetensors",
|
| 218 |
+
"model.layers.2.mlp.router.bias": "model-00001-of-00009.safetensors",
|
| 219 |
+
"model.layers.2.mlp.router.weight": "model-00001-of-00009.safetensors",
|
| 220 |
+
"model.layers.2.post_attention_layernorm.weight": "model-00002-of-00009.safetensors",
|
| 221 |
+
"model.layers.2.self_attn.k_proj.bias": "model-00001-of-00009.safetensors",
|
| 222 |
+
"model.layers.2.self_attn.k_proj.weight": "model-00001-of-00009.safetensors",
|
| 223 |
+
"model.layers.2.self_attn.o_proj.bias": "model-00001-of-00009.safetensors",
|
| 224 |
+
"model.layers.2.self_attn.o_proj.weight": "model-00001-of-00009.safetensors",
|
| 225 |
+
"model.layers.2.self_attn.q_proj.bias": "model-00001-of-00009.safetensors",
|
| 226 |
+
"model.layers.2.self_attn.q_proj.weight": "model-00001-of-00009.safetensors",
|
| 227 |
+
"model.layers.2.self_attn.sinks": "model-00001-of-00009.safetensors",
|
| 228 |
+
"model.layers.2.self_attn.v_proj.bias": "model-00001-of-00009.safetensors",
|
| 229 |
+
"model.layers.2.self_attn.v_proj.weight": "model-00001-of-00009.safetensors",
|
| 230 |
+
"model.layers.20.input_layernorm.weight": "model-00008-of-00009.safetensors",
|
| 231 |
+
"model.layers.20.mlp.experts.down_proj": "model-00008-of-00009.safetensors",
|
| 232 |
+
"model.layers.20.mlp.experts.down_proj_bias": "model-00008-of-00009.safetensors",
|
| 233 |
+
"model.layers.20.mlp.experts.gate_up_proj": "model-00008-of-00009.safetensors",
|
| 234 |
+
"model.layers.20.mlp.experts.gate_up_proj_bias": "model-00008-of-00009.safetensors",
|
| 235 |
+
"model.layers.20.mlp.router.bias": "model-00007-of-00009.safetensors",
|
| 236 |
+
"model.layers.20.mlp.router.weight": "model-00007-of-00009.safetensors",
|
| 237 |
+
"model.layers.20.post_attention_layernorm.weight": "model-00008-of-00009.safetensors",
|
| 238 |
+
"model.layers.20.self_attn.k_proj.bias": "model-00007-of-00009.safetensors",
|
| 239 |
+
"model.layers.20.self_attn.k_proj.weight": "model-00007-of-00009.safetensors",
|
| 240 |
+
"model.layers.20.self_attn.o_proj.bias": "model-00007-of-00009.safetensors",
|
| 241 |
+
"model.layers.20.self_attn.o_proj.weight": "model-00007-of-00009.safetensors",
|
| 242 |
+
"model.layers.20.self_attn.q_proj.bias": "model-00007-of-00009.safetensors",
|
| 243 |
+
"model.layers.20.self_attn.q_proj.weight": "model-00007-of-00009.safetensors",
|
| 244 |
+
"model.layers.20.self_attn.sinks": "model-00007-of-00009.safetensors",
|
| 245 |
+
"model.layers.20.self_attn.v_proj.bias": "model-00007-of-00009.safetensors",
|
| 246 |
+
"model.layers.20.self_attn.v_proj.weight": "model-00007-of-00009.safetensors",
|
| 247 |
+
"model.layers.21.input_layernorm.weight": "model-00008-of-00009.safetensors",
|
| 248 |
+
"model.layers.21.mlp.experts.down_proj": "model-00008-of-00009.safetensors",
|
| 249 |
+
"model.layers.21.mlp.experts.down_proj_bias": "model-00008-of-00009.safetensors",
|
| 250 |
+
"model.layers.21.mlp.experts.gate_up_proj": "model-00008-of-00009.safetensors",
|
| 251 |
+
"model.layers.21.mlp.experts.gate_up_proj_bias": "model-00008-of-00009.safetensors",
|
| 252 |
+
"model.layers.21.mlp.router.bias": "model-00008-of-00009.safetensors",
|
| 253 |
+
"model.layers.21.mlp.router.weight": "model-00008-of-00009.safetensors",
|
| 254 |
+
"model.layers.21.post_attention_layernorm.weight": "model-00008-of-00009.safetensors",
|
| 255 |
+
"model.layers.21.self_attn.k_proj.bias": "model-00008-of-00009.safetensors",
|
| 256 |
+
"model.layers.21.self_attn.k_proj.weight": "model-00008-of-00009.safetensors",
|
| 257 |
+
"model.layers.21.self_attn.o_proj.bias": "model-00008-of-00009.safetensors",
|
| 258 |
+
"model.layers.21.self_attn.o_proj.weight": "model-00008-of-00009.safetensors",
|
| 259 |
+
"model.layers.21.self_attn.q_proj.bias": "model-00008-of-00009.safetensors",
|
| 260 |
+
"model.layers.21.self_attn.q_proj.weight": "model-00008-of-00009.safetensors",
|
| 261 |
+
"model.layers.21.self_attn.sinks": "model-00008-of-00009.safetensors",
|
| 262 |
+
"model.layers.21.self_attn.v_proj.bias": "model-00008-of-00009.safetensors",
|
| 263 |
+
"model.layers.21.self_attn.v_proj.weight": "model-00008-of-00009.safetensors",
|
| 264 |
+
"model.layers.22.input_layernorm.weight": "model-00008-of-00009.safetensors",
|
| 265 |
+
"model.layers.22.mlp.experts.down_proj": "model-00008-of-00009.safetensors",
|
| 266 |
+
"model.layers.22.mlp.experts.down_proj_bias": "model-00008-of-00009.safetensors",
|
| 267 |
+
"model.layers.22.mlp.experts.gate_up_proj": "model-00008-of-00009.safetensors",
|
| 268 |
+
"model.layers.22.mlp.experts.gate_up_proj_bias": "model-00008-of-00009.safetensors",
|
| 269 |
+
"model.layers.22.mlp.router.bias": "model-00008-of-00009.safetensors",
|
| 270 |
+
"model.layers.22.mlp.router.weight": "model-00008-of-00009.safetensors",
|
| 271 |
+
"model.layers.22.post_attention_layernorm.weight": "model-00008-of-00009.safetensors",
|
| 272 |
+
"model.layers.22.self_attn.k_proj.bias": "model-00008-of-00009.safetensors",
|
| 273 |
+
"model.layers.22.self_attn.k_proj.weight": "model-00008-of-00009.safetensors",
|
| 274 |
+
"model.layers.22.self_attn.o_proj.bias": "model-00008-of-00009.safetensors",
|
| 275 |
+
"model.layers.22.self_attn.o_proj.weight": "model-00008-of-00009.safetensors",
|
| 276 |
+
"model.layers.22.self_attn.q_proj.bias": "model-00008-of-00009.safetensors",
|
| 277 |
+
"model.layers.22.self_attn.q_proj.weight": "model-00008-of-00009.safetensors",
|
| 278 |
+
"model.layers.22.self_attn.sinks": "model-00008-of-00009.safetensors",
|
| 279 |
+
"model.layers.22.self_attn.v_proj.bias": "model-00008-of-00009.safetensors",
|
| 280 |
+
"model.layers.22.self_attn.v_proj.weight": "model-00008-of-00009.safetensors",
|
| 281 |
+
"model.layers.23.input_layernorm.weight": "model-00009-of-00009.safetensors",
|
| 282 |
+
"model.layers.23.mlp.experts.down_proj": "model-00009-of-00009.safetensors",
|
| 283 |
+
"model.layers.23.mlp.experts.down_proj_bias": "model-00009-of-00009.safetensors",
|
| 284 |
+
"model.layers.23.mlp.experts.gate_up_proj": "model-00009-of-00009.safetensors",
|
| 285 |
+
"model.layers.23.mlp.experts.gate_up_proj_bias": "model-00009-of-00009.safetensors",
|
| 286 |
+
"model.layers.23.mlp.router.bias": "model-00008-of-00009.safetensors",
|
| 287 |
+
"model.layers.23.mlp.router.weight": "model-00008-of-00009.safetensors",
|
| 288 |
+
"model.layers.23.post_attention_layernorm.weight": "model-00009-of-00009.safetensors",
|
| 289 |
+
"model.layers.23.self_attn.k_proj.bias": "model-00008-of-00009.safetensors",
|
| 290 |
+
"model.layers.23.self_attn.k_proj.weight": "model-00008-of-00009.safetensors",
|
| 291 |
+
"model.layers.23.self_attn.o_proj.bias": "model-00008-of-00009.safetensors",
|
| 292 |
+
"model.layers.23.self_attn.o_proj.weight": "model-00008-of-00009.safetensors",
|
| 293 |
+
"model.layers.23.self_attn.q_proj.bias": "model-00008-of-00009.safetensors",
|
| 294 |
+
"model.layers.23.self_attn.q_proj.weight": "model-00008-of-00009.safetensors",
|
| 295 |
+
"model.layers.23.self_attn.sinks": "model-00008-of-00009.safetensors",
|
| 296 |
+
"model.layers.23.self_attn.v_proj.bias": "model-00008-of-00009.safetensors",
|
| 297 |
+
"model.layers.23.self_attn.v_proj.weight": "model-00008-of-00009.safetensors",
|
| 298 |
+
"model.layers.3.input_layernorm.weight": "model-00002-of-00009.safetensors",
|
| 299 |
+
"model.layers.3.mlp.experts.down_proj": "model-00002-of-00009.safetensors",
|
| 300 |
+
"model.layers.3.mlp.experts.down_proj_bias": "model-00002-of-00009.safetensors",
|
| 301 |
+
"model.layers.3.mlp.experts.gate_up_proj": "model-00002-of-00009.safetensors",
|
| 302 |
+
"model.layers.3.mlp.experts.gate_up_proj_bias": "model-00002-of-00009.safetensors",
|
| 303 |
+
"model.layers.3.mlp.router.bias": "model-00002-of-00009.safetensors",
|
| 304 |
+
"model.layers.3.mlp.router.weight": "model-00002-of-00009.safetensors",
|
| 305 |
+
"model.layers.3.post_attention_layernorm.weight": "model-00002-of-00009.safetensors",
|
| 306 |
+
"model.layers.3.self_attn.k_proj.bias": "model-00002-of-00009.safetensors",
|
| 307 |
+
"model.layers.3.self_attn.k_proj.weight": "model-00002-of-00009.safetensors",
|
| 308 |
+
"model.layers.3.self_attn.o_proj.bias": "model-00002-of-00009.safetensors",
|
| 309 |
+
"model.layers.3.self_attn.o_proj.weight": "model-00002-of-00009.safetensors",
|
| 310 |
+
"model.layers.3.self_attn.q_proj.bias": "model-00002-of-00009.safetensors",
|
| 311 |
+
"model.layers.3.self_attn.q_proj.weight": "model-00002-of-00009.safetensors",
|
| 312 |
+
"model.layers.3.self_attn.sinks": "model-00002-of-00009.safetensors",
|
| 313 |
+
"model.layers.3.self_attn.v_proj.bias": "model-00002-of-00009.safetensors",
|
| 314 |
+
"model.layers.3.self_attn.v_proj.weight": "model-00002-of-00009.safetensors",
|
| 315 |
+
"model.layers.4.input_layernorm.weight": "model-00002-of-00009.safetensors",
|
| 316 |
+
"model.layers.4.mlp.experts.down_proj": "model-00002-of-00009.safetensors",
|
| 317 |
+
"model.layers.4.mlp.experts.down_proj_bias": "model-00002-of-00009.safetensors",
|
| 318 |
+
"model.layers.4.mlp.experts.gate_up_proj": "model-00002-of-00009.safetensors",
|
| 319 |
+
"model.layers.4.mlp.experts.gate_up_proj_bias": "model-00002-of-00009.safetensors",
|
| 320 |
+
"model.layers.4.mlp.router.bias": "model-00002-of-00009.safetensors",
|
| 321 |
+
"model.layers.4.mlp.router.weight": "model-00002-of-00009.safetensors",
|
| 322 |
+
"model.layers.4.post_attention_layernorm.weight": "model-00002-of-00009.safetensors",
|
| 323 |
+
"model.layers.4.self_attn.k_proj.bias": "model-00002-of-00009.safetensors",
|
| 324 |
+
"model.layers.4.self_attn.k_proj.weight": "model-00002-of-00009.safetensors",
|
| 325 |
+
"model.layers.4.self_attn.o_proj.bias": "model-00002-of-00009.safetensors",
|
| 326 |
+
"model.layers.4.self_attn.o_proj.weight": "model-00002-of-00009.safetensors",
|
| 327 |
+
"model.layers.4.self_attn.q_proj.bias": "model-00002-of-00009.safetensors",
|
| 328 |
+
"model.layers.4.self_attn.q_proj.weight": "model-00002-of-00009.safetensors",
|
| 329 |
+
"model.layers.4.self_attn.sinks": "model-00002-of-00009.safetensors",
|
| 330 |
+
"model.layers.4.self_attn.v_proj.bias": "model-00002-of-00009.safetensors",
|
| 331 |
+
"model.layers.4.self_attn.v_proj.weight": "model-00002-of-00009.safetensors",
|
| 332 |
+
"model.layers.5.input_layernorm.weight": "model-00003-of-00009.safetensors",
|
| 333 |
+
"model.layers.5.mlp.experts.down_proj": "model-00003-of-00009.safetensors",
|
| 334 |
+
"model.layers.5.mlp.experts.down_proj_bias": "model-00003-of-00009.safetensors",
|
| 335 |
+
"model.layers.5.mlp.experts.gate_up_proj": "model-00003-of-00009.safetensors",
|
| 336 |
+
"model.layers.5.mlp.experts.gate_up_proj_bias": "model-00003-of-00009.safetensors",
|
| 337 |
+
"model.layers.5.mlp.router.bias": "model-00002-of-00009.safetensors",
|
| 338 |
+
"model.layers.5.mlp.router.weight": "model-00002-of-00009.safetensors",
|
| 339 |
+
"model.layers.5.post_attention_layernorm.weight": "model-00003-of-00009.safetensors",
|
| 340 |
+
"model.layers.5.self_attn.k_proj.bias": "model-00002-of-00009.safetensors",
|
| 341 |
+
"model.layers.5.self_attn.k_proj.weight": "model-00002-of-00009.safetensors",
|
| 342 |
+
"model.layers.5.self_attn.o_proj.bias": "model-00002-of-00009.safetensors",
|
| 343 |
+
"model.layers.5.self_attn.o_proj.weight": "model-00002-of-00009.safetensors",
|
| 344 |
+
"model.layers.5.self_attn.q_proj.bias": "model-00002-of-00009.safetensors",
|
| 345 |
+
"model.layers.5.self_attn.q_proj.weight": "model-00002-of-00009.safetensors",
|
| 346 |
+
"model.layers.5.self_attn.sinks": "model-00002-of-00009.safetensors",
|
| 347 |
+
"model.layers.5.self_attn.v_proj.bias": "model-00002-of-00009.safetensors",
|
| 348 |
+
"model.layers.5.self_attn.v_proj.weight": "model-00002-of-00009.safetensors",
|
| 349 |
+
"model.layers.6.input_layernorm.weight": "model-00003-of-00009.safetensors",
|
| 350 |
+
"model.layers.6.mlp.experts.down_proj": "model-00003-of-00009.safetensors",
|
| 351 |
+
"model.layers.6.mlp.experts.down_proj_bias": "model-00003-of-00009.safetensors",
|
| 352 |
+
"model.layers.6.mlp.experts.gate_up_proj": "model-00003-of-00009.safetensors",
|
| 353 |
+
"model.layers.6.mlp.experts.gate_up_proj_bias": "model-00003-of-00009.safetensors",
|
| 354 |
+
"model.layers.6.mlp.router.bias": "model-00003-of-00009.safetensors",
|
| 355 |
+
"model.layers.6.mlp.router.weight": "model-00003-of-00009.safetensors",
|
| 356 |
+
"model.layers.6.post_attention_layernorm.weight": "model-00003-of-00009.safetensors",
|
| 357 |
+
"model.layers.6.self_attn.k_proj.bias": "model-00003-of-00009.safetensors",
|
| 358 |
+
"model.layers.6.self_attn.k_proj.weight": "model-00003-of-00009.safetensors",
|
| 359 |
+
"model.layers.6.self_attn.o_proj.bias": "model-00003-of-00009.safetensors",
|
| 360 |
+
"model.layers.6.self_attn.o_proj.weight": "model-00003-of-00009.safetensors",
|
| 361 |
+
"model.layers.6.self_attn.q_proj.bias": "model-00003-of-00009.safetensors",
|
| 362 |
+
"model.layers.6.self_attn.q_proj.weight": "model-00003-of-00009.safetensors",
|
| 363 |
+
"model.layers.6.self_attn.sinks": "model-00003-of-00009.safetensors",
|
| 364 |
+
"model.layers.6.self_attn.v_proj.bias": "model-00003-of-00009.safetensors",
|
| 365 |
+
"model.layers.6.self_attn.v_proj.weight": "model-00003-of-00009.safetensors",
|
| 366 |
+
"model.layers.7.input_layernorm.weight": "model-00003-of-00009.safetensors",
|
| 367 |
+
"model.layers.7.mlp.experts.down_proj": "model-00003-of-00009.safetensors",
|
| 368 |
+
"model.layers.7.mlp.experts.down_proj_bias": "model-00003-of-00009.safetensors",
|
| 369 |
+
"model.layers.7.mlp.experts.gate_up_proj": "model-00003-of-00009.safetensors",
|
| 370 |
+
"model.layers.7.mlp.experts.gate_up_proj_bias": "model-00003-of-00009.safetensors",
|
| 371 |
+
"model.layers.7.mlp.router.bias": "model-00003-of-00009.safetensors",
|
| 372 |
+
"model.layers.7.mlp.router.weight": "model-00003-of-00009.safetensors",
|
| 373 |
+
"model.layers.7.post_attention_layernorm.weight": "model-00003-of-00009.safetensors",
|
| 374 |
+
"model.layers.7.self_attn.k_proj.bias": "model-00003-of-00009.safetensors",
|
| 375 |
+
"model.layers.7.self_attn.k_proj.weight": "model-00003-of-00009.safetensors",
|
| 376 |
+
"model.layers.7.self_attn.o_proj.bias": "model-00003-of-00009.safetensors",
|
| 377 |
+
"model.layers.7.self_attn.o_proj.weight": "model-00003-of-00009.safetensors",
|
| 378 |
+
"model.layers.7.self_attn.q_proj.bias": "model-00003-of-00009.safetensors",
|
| 379 |
+
"model.layers.7.self_attn.q_proj.weight": "model-00003-of-00009.safetensors",
|
| 380 |
+
"model.layers.7.self_attn.sinks": "model-00003-of-00009.safetensors",
|
| 381 |
+
"model.layers.7.self_attn.v_proj.bias": "model-00003-of-00009.safetensors",
|
| 382 |
+
"model.layers.7.self_attn.v_proj.weight": "model-00003-of-00009.safetensors",
|
| 383 |
+
"model.layers.8.input_layernorm.weight": "model-00004-of-00009.safetensors",
|
| 384 |
+
"model.layers.8.mlp.experts.down_proj": "model-00004-of-00009.safetensors",
|
| 385 |
+
"model.layers.8.mlp.experts.down_proj_bias": "model-00004-of-00009.safetensors",
|
| 386 |
+
"model.layers.8.mlp.experts.gate_up_proj": "model-00004-of-00009.safetensors",
|
| 387 |
+
"model.layers.8.mlp.experts.gate_up_proj_bias": "model-00004-of-00009.safetensors",
|
| 388 |
+
"model.layers.8.mlp.router.bias": "model-00003-of-00009.safetensors",
|
| 389 |
+
"model.layers.8.mlp.router.weight": "model-00003-of-00009.safetensors",
|
| 390 |
+
"model.layers.8.post_attention_layernorm.weight": "model-00004-of-00009.safetensors",
|
| 391 |
+
"model.layers.8.self_attn.k_proj.bias": "model-00003-of-00009.safetensors",
|
| 392 |
+
"model.layers.8.self_attn.k_proj.weight": "model-00003-of-00009.safetensors",
|
| 393 |
+
"model.layers.8.self_attn.o_proj.bias": "model-00003-of-00009.safetensors",
|
| 394 |
+
"model.layers.8.self_attn.o_proj.weight": "model-00003-of-00009.safetensors",
|
| 395 |
+
"model.layers.8.self_attn.q_proj.bias": "model-00003-of-00009.safetensors",
|
| 396 |
+
"model.layers.8.self_attn.q_proj.weight": "model-00003-of-00009.safetensors",
|
| 397 |
+
"model.layers.8.self_attn.sinks": "model-00003-of-00009.safetensors",
|
| 398 |
+
"model.layers.8.self_attn.v_proj.bias": "model-00003-of-00009.safetensors",
|
| 399 |
+
"model.layers.8.self_attn.v_proj.weight": "model-00003-of-00009.safetensors",
|
| 400 |
+
"model.layers.9.input_layernorm.weight": "model-00004-of-00009.safetensors",
|
| 401 |
+
"model.layers.9.mlp.experts.down_proj": "model-00004-of-00009.safetensors",
|
| 402 |
+
"model.layers.9.mlp.experts.down_proj_bias": "model-00004-of-00009.safetensors",
|
| 403 |
+
"model.layers.9.mlp.experts.gate_up_proj": "model-00004-of-00009.safetensors",
|
| 404 |
+
"model.layers.9.mlp.experts.gate_up_proj_bias": "model-00004-of-00009.safetensors",
|
| 405 |
+
"model.layers.9.mlp.router.bias": "model-00004-of-00009.safetensors",
|
| 406 |
+
"model.layers.9.mlp.router.weight": "model-00004-of-00009.safetensors",
|
| 407 |
+
"model.layers.9.post_attention_layernorm.weight": "model-00004-of-00009.safetensors",
|
| 408 |
+
"model.layers.9.self_attn.k_proj.bias": "model-00004-of-00009.safetensors",
|
| 409 |
+
"model.layers.9.self_attn.k_proj.weight": "model-00004-of-00009.safetensors",
|
| 410 |
+
"model.layers.9.self_attn.o_proj.bias": "model-00004-of-00009.safetensors",
|
| 411 |
+
"model.layers.9.self_attn.o_proj.weight": "model-00004-of-00009.safetensors",
|
| 412 |
+
"model.layers.9.self_attn.q_proj.bias": "model-00004-of-00009.safetensors",
|
| 413 |
+
"model.layers.9.self_attn.q_proj.weight": "model-00004-of-00009.safetensors",
|
| 414 |
+
"model.layers.9.self_attn.sinks": "model-00004-of-00009.safetensors",
|
| 415 |
+
"model.layers.9.self_attn.v_proj.bias": "model-00004-of-00009.safetensors",
|
| 416 |
+
"model.layers.9.self_attn.v_proj.weight": "model-00004-of-00009.safetensors",
|
| 417 |
+
"model.norm.weight": "model-00009-of-00009.safetensors"
|
| 418 |
+
}
|
| 419 |
+
}
|
model_card.json
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_name": "MGI Marketing Intelligence Model",
|
| 3 |
+
"base_model": "gpt-oss-20b-full",
|
| 4 |
+
"training_stages": [
|
| 5 |
+
"Initial SFT training (984 examples)",
|
| 6 |
+
"Enhanced training with synthetic data (994 examples)",
|
| 7 |
+
"Surgical fine-tuning (50 examples, 2 epochs)"
|
| 8 |
+
],
|
| 9 |
+
"merge_timestamp": "2025-08-07 04:00:00.358600",
|
| 10 |
+
"benchmark_score": "79.5% on MGI Benchmark",
|
| 11 |
+
"advanced_benchmark": "75.3% on Advanced MGI Benchmark",
|
| 12 |
+
"production_ready": true,
|
| 13 |
+
"inference_parameters": {
|
| 14 |
+
"temperature": 0.7,
|
| 15 |
+
"top_p": 0.9,
|
| 16 |
+
"repetition_penalty": 1.1,
|
| 17 |
+
"max_new_tokens": 400,
|
| 18 |
+
"min_new_tokens": 100
|
| 19 |
+
}
|
| 20 |
+
}
|
special_tokens_map.json
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"bos_token": {
|
| 3 |
+
"content": "<|startoftext|>",
|
| 4 |
+
"lstrip": false,
|
| 5 |
+
"normalized": false,
|
| 6 |
+
"rstrip": false,
|
| 7 |
+
"single_word": false
|
| 8 |
+
},
|
| 9 |
+
"eos_token": {
|
| 10 |
+
"content": "<|return|>",
|
| 11 |
+
"lstrip": false,
|
| 12 |
+
"normalized": false,
|
| 13 |
+
"rstrip": false,
|
| 14 |
+
"single_word": false
|
| 15 |
+
},
|
| 16 |
+
"pad_token": {
|
| 17 |
+
"content": "<|endoftext|>",
|
| 18 |
+
"lstrip": false,
|
| 19 |
+
"normalized": false,
|
| 20 |
+
"rstrip": false,
|
| 21 |
+
"single_word": false
|
| 22 |
+
}
|
| 23 |
+
}
|
tokenizer.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0614fe83cadab421296e664e1f48f4261fa8fef6e03e63bb75c20f38e37d07d3
|
| 3 |
+
size 27868174
|
tokenizer_config.json
ADDED
|
@@ -0,0 +1,183 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"added_tokens_decoder": {
|
| 3 |
+
"199998": {
|
| 4 |
+
"content": "<|startoftext|>",
|
| 5 |
+
"lstrip": false,
|
| 6 |
+
"normalized": false,
|
| 7 |
+
"rstrip": false,
|
| 8 |
+
"single_word": false,
|
| 9 |
+
"special": true
|
| 10 |
+
},
|
| 11 |
+
"199999": {
|
| 12 |
+
"content": "<|endoftext|>",
|
| 13 |
+
"lstrip": false,
|
| 14 |
+
"normalized": false,
|
| 15 |
+
"rstrip": false,
|
| 16 |
+
"single_word": false,
|
| 17 |
+
"special": true
|
| 18 |
+
},
|
| 19 |
+
"200000": {
|
| 20 |
+
"content": "<|reserved_200000|>",
|
| 21 |
+
"lstrip": false,
|
| 22 |
+
"normalized": false,
|
| 23 |
+
"rstrip": false,
|
| 24 |
+
"single_word": false,
|
| 25 |
+
"special": true
|
| 26 |
+
},
|
| 27 |
+
"200001": {
|
| 28 |
+
"content": "<|reserved_200001|>",
|
| 29 |
+
"lstrip": false,
|
| 30 |
+
"normalized": false,
|
| 31 |
+
"rstrip": false,
|
| 32 |
+
"single_word": false,
|
| 33 |
+
"special": true
|
| 34 |
+
},
|
| 35 |
+
"200002": {
|
| 36 |
+
"content": "<|return|>",
|
| 37 |
+
"lstrip": false,
|
| 38 |
+
"normalized": false,
|
| 39 |
+
"rstrip": false,
|
| 40 |
+
"single_word": false,
|
| 41 |
+
"special": true
|
| 42 |
+
},
|
| 43 |
+
"200003": {
|
| 44 |
+
"content": "<|constrain|>",
|
| 45 |
+
"lstrip": false,
|
| 46 |
+
"normalized": false,
|
| 47 |
+
"rstrip": false,
|
| 48 |
+
"single_word": false,
|
| 49 |
+
"special": true
|
| 50 |
+
},
|
| 51 |
+
"200004": {
|
| 52 |
+
"content": "<|reserved_200004|>",
|
| 53 |
+
"lstrip": false,
|
| 54 |
+
"normalized": false,
|
| 55 |
+
"rstrip": false,
|
| 56 |
+
"single_word": false,
|
| 57 |
+
"special": true
|
| 58 |
+
},
|
| 59 |
+
"200005": {
|
| 60 |
+
"content": "<|channel|>",
|
| 61 |
+
"lstrip": false,
|
| 62 |
+
"normalized": false,
|
| 63 |
+
"rstrip": false,
|
| 64 |
+
"single_word": false,
|
| 65 |
+
"special": true
|
| 66 |
+
},
|
| 67 |
+
"200006": {
|
| 68 |
+
"content": "<|start|>",
|
| 69 |
+
"lstrip": false,
|
| 70 |
+
"normalized": false,
|
| 71 |
+
"rstrip": false,
|
| 72 |
+
"single_word": false,
|
| 73 |
+
"special": true
|
| 74 |
+
},
|
| 75 |
+
"200007": {
|
| 76 |
+
"content": "<|end|>",
|
| 77 |
+
"lstrip": false,
|
| 78 |
+
"normalized": false,
|
| 79 |
+
"rstrip": false,
|
| 80 |
+
"single_word": false,
|
| 81 |
+
"special": true
|
| 82 |
+
},
|
| 83 |
+
"200008": {
|
| 84 |
+
"content": "<|message|>",
|
| 85 |
+
"lstrip": false,
|
| 86 |
+
"normalized": false,
|
| 87 |
+
"rstrip": false,
|
| 88 |
+
"single_word": false,
|
| 89 |
+
"special": true
|
| 90 |
+
},
|
| 91 |
+
"200009": {
|
| 92 |
+
"content": "<|reserved_200009|>",
|
| 93 |
+
"lstrip": false,
|
| 94 |
+
"normalized": false,
|
| 95 |
+
"rstrip": false,
|
| 96 |
+
"single_word": false,
|
| 97 |
+
"special": true
|
| 98 |
+
},
|
| 99 |
+
"200010": {
|
| 100 |
+
"content": "<|reserved_200010|>",
|
| 101 |
+
"lstrip": false,
|
| 102 |
+
"normalized": false,
|
| 103 |
+
"rstrip": false,
|
| 104 |
+
"single_word": false,
|
| 105 |
+
"special": true
|
| 106 |
+
},
|
| 107 |
+
"200011": {
|
| 108 |
+
"content": "<|reserved_200011|>",
|
| 109 |
+
"lstrip": false,
|
| 110 |
+
"normalized": false,
|
| 111 |
+
"rstrip": false,
|
| 112 |
+
"single_word": false,
|
| 113 |
+
"special": true
|
| 114 |
+
},
|
| 115 |
+
"200012": {
|
| 116 |
+
"content": "<|call|>",
|
| 117 |
+
"lstrip": false,
|
| 118 |
+
"normalized": false,
|
| 119 |
+
"rstrip": false,
|
| 120 |
+
"single_word": false,
|
| 121 |
+
"special": true
|
| 122 |
+
},
|
| 123 |
+
"200013": {
|
| 124 |
+
"content": "<|reserved_200013|>",
|
| 125 |
+
"lstrip": false,
|
| 126 |
+
"normalized": false,
|
| 127 |
+
"rstrip": false,
|
| 128 |
+
"single_word": false,
|
| 129 |
+
"special": true
|
| 130 |
+
},
|
| 131 |
+
"200014": {
|
| 132 |
+
"content": "<|reserved_200014|>",
|
| 133 |
+
"lstrip": false,
|
| 134 |
+
"normalized": false,
|
| 135 |
+
"rstrip": false,
|
| 136 |
+
"single_word": false,
|
| 137 |
+
"special": true
|
| 138 |
+
},
|
| 139 |
+
"200015": {
|
| 140 |
+
"content": "<|reserved_200015|>",
|
| 141 |
+
"lstrip": false,
|
| 142 |
+
"normalized": false,
|
| 143 |
+
"rstrip": false,
|
| 144 |
+
"single_word": false,
|
| 145 |
+
"special": true
|
| 146 |
+
},
|
| 147 |
+
"200016": {
|
| 148 |
+
"content": "<|reserved_200016|>",
|
| 149 |
+
"lstrip": false,
|
| 150 |
+
"normalized": false,
|
| 151 |
+
"rstrip": false,
|
| 152 |
+
"single_word": false,
|
| 153 |
+
"special": true
|
| 154 |
+
},
|
| 155 |
+
"200017": {
|
| 156 |
+
"content": "<|reserved_200017|>",
|
| 157 |
+
"lstrip": false,
|
| 158 |
+
"normalized": false,
|
| 159 |
+
"rstrip": false,
|
| 160 |
+
"single_word": false,
|
| 161 |
+
"special": true
|
| 162 |
+
},
|
| 163 |
+
"200018": {
|
| 164 |
+
"content": "<|endofprompt|>",
|
| 165 |
+
"lstrip": false,
|
| 166 |
+
"normalized": false,
|
| 167 |
+
"rstrip": false,
|
| 168 |
+
"single_word": false,
|
| 169 |
+
"special": true
|
| 170 |
+
}
|
| 171 |
+
},
|
| 172 |
+
"bos_token": "<|startoftext|>",
|
| 173 |
+
"clean_up_tokenization_spaces": false,
|
| 174 |
+
"eos_token": "<|return|>",
|
| 175 |
+
"extra_special_tokens": {},
|
| 176 |
+
"model_input_names": [
|
| 177 |
+
"input_ids",
|
| 178 |
+
"attention_mask"
|
| 179 |
+
],
|
| 180 |
+
"model_max_length": 1000000000000000019884624838656,
|
| 181 |
+
"pad_token": "<|endoftext|>",
|
| 182 |
+
"tokenizer_class": "PreTrainedTokenizerFast"
|
| 183 |
+
}
|