Update README.md
Browse files
README.md
CHANGED
|
@@ -8,6 +8,108 @@ pipeline_tag: text-generation
|
|
| 8 |
library_name: transformers
|
| 9 |
tags:
|
| 10 |
- text-generation-inference
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
---
|
| 12 |
|
| 13 |
-

|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
library_name: transformers
|
| 9 |
tags:
|
| 10 |
- text-generation-inference
|
| 11 |
+
- abliterated
|
| 12 |
+
- math
|
| 13 |
+
- moderately abliterated
|
| 14 |
+
- code
|
| 15 |
---
|
| 16 |
|
| 17 |
+

|
| 18 |
+
|
| 19 |
+
# **Sombrero-R1-14B-Elite13**
|
| 20 |
+
|
| 21 |
+
> Sombrero-R1-14B-Elite13 is a fine-tuned variant of the DeepSeek-R1-Distill-Qwen-14B model, enhanced through reinforcement learning to serve as a high-performance reasoning assistant. It excels in both mathematical problem-solving and general-purpose conversational tasks. This model combines distilled efficiency with refined instruction-following behavior, offering an ideal balance of speed, capability, and coherence for complex interactive tasks.
|
| 22 |
+
|
| 23 |
+
### Key Enhancements
|
| 24 |
+
|
| 25 |
+
1. **Reinforcement Learning Fine-Tuning**
|
| 26 |
+
Trained with reinforcement learning objectives to optimize for alignment, reward-guided reasoning, and helpfulness in conversation.
|
| 27 |
+
|
| 28 |
+
2. **Mathematical Reasoning Proficiency**
|
| 29 |
+
Delivers accurate solutions and step-by-step breakdowns for algebra, calculus, number theory, logic puzzles, and applied mathematics.
|
| 30 |
+
|
| 31 |
+
3. **Instruction Adherence**
|
| 32 |
+
Capable of understanding and following multi-part instructions, including structured tasks and iterative refinement prompts.
|
| 33 |
+
|
| 34 |
+
4. **Expanded Context Handling**
|
| 35 |
+
Supports up to 128K tokens of context with output lengths up to 8K tokens, ideal for technical and educational use cases.
|
| 36 |
+
|
| 37 |
+
5. **Cross-Domain Knowledge**
|
| 38 |
+
Offers broad general knowledge capabilities, making it suitable for tutoring, research, and exploratory conversation across topics.
|
| 39 |
+
|
| 40 |
+
---
|
| 41 |
+
|
| 42 |
+
# **Quickstart with Transformers**
|
| 43 |
+
|
| 44 |
+
```python
|
| 45 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 46 |
+
|
| 47 |
+
model_name = "prithivMLmods/Sombrero-R1-14B-Elite13"
|
| 48 |
+
|
| 49 |
+
model = AutoModelForCausalLM.from_pretrained(
|
| 50 |
+
model_name,
|
| 51 |
+
torch_dtype="auto",
|
| 52 |
+
device_map="auto"
|
| 53 |
+
)
|
| 54 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 55 |
+
|
| 56 |
+
prompt = "Solve: Integrate (x^2 * e^x) dx"
|
| 57 |
+
messages = [
|
| 58 |
+
{"role": "system", "content": "You are a helpful AI assistant skilled in math and reasoning."},
|
| 59 |
+
{"role": "user", "content": prompt}
|
| 60 |
+
]
|
| 61 |
+
text = tokenizer.apply_chat_template(
|
| 62 |
+
messages,
|
| 63 |
+
tokenize=False,
|
| 64 |
+
add_generation_prompt=True
|
| 65 |
+
)
|
| 66 |
+
model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
|
| 67 |
+
|
| 68 |
+
generated_ids = model.generate(
|
| 69 |
+
**model_inputs,
|
| 70 |
+
max_new_tokens=512
|
| 71 |
+
)
|
| 72 |
+
generated_ids = [
|
| 73 |
+
output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
|
| 74 |
+
]
|
| 75 |
+
|
| 76 |
+
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
| 77 |
+
```
|
| 78 |
+
|
| 79 |
+
---
|
| 80 |
+
|
| 81 |
+
# **Intended Use Cases**
|
| 82 |
+
|
| 83 |
+
1. **Mathematics Problem Solving**
|
| 84 |
+
Ideal for step-by-step derivations, symbolic computation, numerical explanations, and LaTeX-supported outputs.
|
| 85 |
+
|
| 86 |
+
2. **Educational and Instructional Support**
|
| 87 |
+
Helpful in classrooms and learning platforms, offering guided explanations for students and instructors.
|
| 88 |
+
|
| 89 |
+
3. **Chat-based Reasoning**
|
| 90 |
+
Designed for coherent, context-aware dialogue generation with structured logic and continuity.
|
| 91 |
+
|
| 92 |
+
4. **Multilingual Knowledge Assistance**
|
| 93 |
+
Supports 29+ languages, including English, Chinese, French, German, Arabic, and others, for multilingual learning.
|
| 94 |
+
|
| 95 |
+
5. **Document and Code Explanation**
|
| 96 |
+
Can explain complex documents, code snippets, or structured logic flows in natural language.
|
| 97 |
+
|
| 98 |
+
---
|
| 99 |
+
|
| 100 |
+
# **Known Limitations**
|
| 101 |
+
|
| 102 |
+
1. **Compute Intensive**
|
| 103 |
+
Requires high-memory hardware (e.g., ≥48GB VRAM) to fully utilize context length and generation capacity.
|
| 104 |
+
|
| 105 |
+
2. **Potential for Bias and Hallucinations**
|
| 106 |
+
While tuned for alignment, some responses may still exhibit artifacts from pretraining biases or inaccuracies in edge cases.
|
| 107 |
+
|
| 108 |
+
3. **Drift in Long Responses**
|
| 109 |
+
Output may occasionally degrade in structure or accuracy across long generations.
|
| 110 |
+
|
| 111 |
+
4. **Static Knowledge**
|
| 112 |
+
Does not have real-time awareness or access to events or research developments post-training.
|
| 113 |
+
|
| 114 |
+
5. **Creative Task Variability**
|
| 115 |
+
While optimized for logic, its performance in narrative or subjective content may be inconsistent.
|