prithivMLmods commited on
Commit
a928116
·
verified ·
1 Parent(s): 23e4d83

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +93 -1
README.md CHANGED
@@ -11,4 +11,96 @@ tags:
11
  - llama3.1
12
  - phi4
13
  - LlamaForCausalLM
14
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  - llama3.1
12
  - phi4
13
  - LlamaForCausalLM
14
+ ---
15
+ # **Megatron-Opus-14B-2.0 [ Exp ]**
16
+
17
+ [Megatron-Opus-14B-2.0 ] Exp finetuned from Microsoft's Phi-4 is a state-of-the-art open model developed with a focus on responsible problem solving and advanced reasoning capabilities. Built upon a diverse blend of synthetic datasets, carefully filtered public domain websites, and high-quality academic books and Q&A datasets, Megatron-Opus-14B-2.0 ensures that small, capable models are trained with datasets of exceptional depth and precision.
18
+
19
+ Megatron-Opus-14B-2.0 adopts a robust safety post-training approach using open-source and in-house synthetic datasets. This involves a combination of SFT (Supervised Fine-Tuning) and iterative DPO (Direct Preference Optimization) techniques, ensuring helpful and harmless outputs across various safety categories.
20
+
21
+
22
+ # **Dataset Info**
23
+
24
+ Megatron-Opus-14B-2.0 is fine-tuned on a carefully curated synthetic dataset generated using an advanced pipeline optimized for Chain of Thought (CoT) reasoning and Responsible Problem Breakdown (RPB) methodologies. This ensures that the model excels at:
25
+
26
+ - **Logical reasoning**
27
+ - **Step-by-step problem-solving**
28
+ - **Breaking down complex tasks into manageable parts**
29
+
30
+ The dataset also emphasizes responsible decision-making and fairness in generating solutions.
31
+
32
+ # **Run with Transformers**
33
+
34
+ ```python
35
+ # pip install accelerate
36
+ from transformers import AutoTokenizer, AutoModelForCausalLM
37
+ import torch
38
+
39
+ tokenizer = AutoTokenizer.from_pretrained("prithivMLmods/Megatron-Opus-14B-2.0")
40
+ model = AutoModelForCausalLM.from_pretrained(
41
+ "prithivMLmods/Megatron-Opus-14B-2.0",
42
+ device_map="auto",
43
+ torch_dtype=torch.bfloat16,
44
+ )
45
+
46
+ input_text = "Explain the concept of black holes."
47
+ input_ids = tokenizer(input_text, return_tensors="pt").to("cuda")
48
+
49
+ outputs = model.generate(**input_ids, max_new_tokens=64)
50
+ print(tokenizer.decode(outputs[0]))
51
+ ```
52
+
53
+ For chat-style interactions, use `tokenizer.apply_chat_template`:
54
+
55
+ ```python
56
+ messages = [
57
+ {"role": "user", "content": "Explain the concept of black holes."},
58
+ ]
59
+ input_ids = tokenizer.apply_chat_template(messages, return_tensors="pt", return_dict=True).to("cuda")
60
+
61
+ outputs = model.generate(**input_ids, max_new_tokens=256)
62
+ print(tokenizer.decode(outputs[0]))
63
+ ```
64
+
65
+ # **Intended Use**
66
+
67
+ Megatron-Opus-14B-2.0 is tailored for a wide range of applications, especially those involving **advanced reasoning**, **multilingual capabilities**, and **responsible problem-solving**. Its primary use cases include:
68
+
69
+ 1. **Responsible Problem Solving**
70
+ - Breaking down complex problems into logical, actionable steps.
71
+ - Offering ethical, well-rounded solutions in academic and professional contexts.
72
+
73
+ 2. **Advanced Reasoning Tasks**
74
+ - Excelling in mathematics, logic, and scientific reasoning.
75
+ - Providing detailed explanations and systematic answers.
76
+
77
+ 3. **Content Generation**
78
+ - Assisting in generating high-quality content for various domains, including creative writing and technical documentation.
79
+ - Supporting marketers, writers, and educators with detailed and well-structured outputs.
80
+
81
+ 4. **Educational Support**
82
+ - Acting as a virtual tutor for students by generating practice questions, answers, and detailed explanations.
83
+ - Helping educators design learning material that promotes critical thinking and step-by-step problem-solving.
84
+
85
+ 5. **Customer Support & Dialogue Systems**
86
+ - Enabling chatbots and virtual assistants to provide accurate, helpful, and responsible responses.
87
+ - Enhancing customer service with reasoning-driven automation.
88
+
89
+ # **Limitations**
90
+
91
+ Despite its strengths, Megatron-Opus-14B-2.0 has some limitations that users should be aware of:
92
+
93
+ 1. **Bias and Fairness**
94
+ - While great effort has been made to minimize biases, users should critically assess the model’s output in sensitive scenarios to avoid unintended bias.
95
+
96
+ 2. **Contextual Interpretation**
97
+ - The model may occasionally misinterpret highly nuanced prompts or ambiguous contexts, leading to suboptimal responses.
98
+
99
+ 3. **Knowledge Cutoff**
100
+ - Megatron-Opus-14B-2.0’s knowledge is static and based on the data available at the time of training. It does not include real-time updates or information on recent developments.
101
+
102
+ 4. **Safety and Harmlessness**
103
+ - Despite post-training safety alignment, inappropriate or harmful outputs may still occur. Continuous monitoring and human oversight are advised when using the model in critical contexts.
104
+
105
+ 5. **Computational Requirements**
106
+ - Deploying Megatron-Opus-14B-2.0 efficiently may require substantial computational resources, particularly for large-scale deployments or real-time applications.