Update README.md

README.md

@@ -19,7 +19,9 @@ tags:
 - safetensors
 - Question Answering
 ---
-### **Llama-3.1-8B-Open-SFT**
+### Llama-3.1-8B-Open-SFT
+
+The **Llama-3.1-8B-Open-SFT** model is a fine-tuned version of **meta-llama/Llama-3.1-8B-Instruct**, designed for advanced text generation tasks, including conversational interactions, question answering, and chain-of-thought reasoning. This model leverages **Supervised Fine-Tuning (SFT)** using the **O1-OPEN/OpenO1-SFT** dataset to provide enhanced performance in context-sensitive and instruction-following tasks.
 
 | **File Name**                       | **Size**          | **Description**                                     | **Upload Status** |
 |-------------------------------------|-------------------|---------------------------------------------------|-------------------|
@@ -36,4 +38,95 @@ tags:
 | `tokenizer.json`                    | 17.2 MB          | Full tokenizer JSON file.                         | Uploaded (LFS)    |
 | `tokenizer_config.json`             | 57.4 kB          | Configuration for the tokenizer.                  | Uploaded          |
 
----
+---
+
+### **Key Features**
+
+1. **Text Generation with CoT Reasoning:**  
+   - Implements **Chain-of-Thought (CoT)** prompting for logical and step-by-step reasoning tasks.
+
+2. **Conversational AI:**  
+   - Excels in generating context-aware and coherent responses in multi-turn conversations.
+
+3. **Supervised Fine-Tuning (SFT):**  
+   - Optimized for open-domain tasks using the **O1-OPEN/OpenO1-SFT** dataset.
+
+4. **Multi-Purpose Functionality:**  
+   - Supports a wide range of NLP tasks, including summarization, question answering, and text completion.
+
+5. **Scalable Sharded Architecture:**  
+   - Model weights are distributed across four shards, ensuring efficient loading for large-scale applications.
+
+---
+
+### **Training Details**
+
+- **Base Model:** [meta-llama/Llama-3.1-8B](#)  
+- **Finetuned Dataset:** [O1-OPEN/OpenO1-SFT](#)  
+   - Dataset includes **77.7k** fine-tuning samples, curated for instruction-based and open-domain tasks.
+
+- **Model Size:**  
+   - 8 Billion parameters distributed over 4 shards for efficient deployment.
+
+### **Applications**
+
+1. **Chain-of-Thought (CoT) Reasoning:**  
+   - Solve complex problems step-by-step with logical reasoning capabilities.
+
+2. **Conversational Agents:**  
+   - Ideal for chatbots, virtual assistants, and conversational systems.
+
+3. **Question Answering:**  
+   - Answer open-domain or context-specific questions accurately.
+
+4. **Text Completion:**  
+   - Generate coherent continuations for incomplete inputs.
+
+5. **Creative Writing:**  
+   - Support for generating stories, articles, or brainstorming ideas.
+
+---
+
+### **Usage**
+
+#### **Loading the Model**
+```python
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+model_name = "prithivMLmods/Llama-3.1-8B-Open-SFT"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+```
+
+---
+
+#### **Inference Example**
+```python
+prompt = """
+Explain the concept of gravity in a simple way suitable for a 10-year-old:
+"""
+inputs = tokenizer(prompt, return_tensors="pt")
+outputs = model.generate(**inputs, max_length=150, temperature=0.7)
+
+response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+print("Model Output:", response)
+```
+
+---
+
+### **Expected Output**
+**"Gravity is a force that pulls things toward each other. It's the reason why things fall to the ground when you drop them. On Earth, gravity keeps us on the ground and makes sure everything stays in place, like your toys, the water in the ocean, and even the air we breathe."**
+
+---
+
+### **Performance Requirements**
+
+- **Hardware:**  
+   - High-performance GPUs are recommended for efficient inference.  
+   - Minimum memory: ~16GB VRAM for full precision; 8GB for quantized models.
+
+- **Optimization Options:**  
+   - Use `Safetensors` for secure and efficient weight loading.  
+   - Quantization or model parallelism for resource-constrained environments.
+
+---