Update README.md
Browse files
README.md
CHANGED
|
@@ -17,4 +17,69 @@ datasets:
|
|
| 17 |
- smirki/UI_REASONING_v1.01
|
| 18 |
- smirki/Parkytest
|
| 19 |
---
|
| 20 |
-

|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
- smirki/UI_REASONING_v1.01
|
| 18 |
- smirki/Parkytest
|
| 19 |
---
|
| 20 |
+

|
| 21 |
+
|
| 22 |
+
# **Viper-OneCoder-UIGEN**
|
| 23 |
+
|
| 24 |
+
Viper-OneCoder-UIGEN is based on the Qwen 2.5 14B modality architecture, designed to be the **best** for web development and structured coding logic. It has been fine-tuned on a synthetic dataset leveraging the latest coding logits and CoT datasets, further optimizing its **step-by-step logic breakdown** and **front-end problem-solving** abilities. The model demonstrates significant improvements in **context understanding, structured UI development, and long-context comprehension**, making it ideal for **web-based coding tasks, HTML/CSS/Tailwind development, and detailed instruction following**.
|
| 25 |
+
|
| 26 |
+
### **Key Improvements**
|
| 27 |
+
1. **Best-in-Class Web Development Proficiency**: Advanced understanding of **HTML, CSS, Tailwind, JavaScript**, and front-end frameworks.
|
| 28 |
+
2. **Fine-Tuned Step-by-Step Logic Breakdown**: Optimized for structured explanations, component-based UI coding, and logic-driven development.
|
| 29 |
+
3. **Advanced Instruction Following**: Delivers precise responses, structured outputs (e.g., JSON, YAML), and extended text generation (**8K+ tokens**).
|
| 30 |
+
4. **Long-Context Mastery**: Handles up to **128K tokens** with an output capability of **8K tokens** per response.
|
| 31 |
+
5. **Multilingual Code Support**: Excels in **HTML, CSS, JavaScript, React, Tailwind CSS, Python**, and other major web-related languages, with documentation in **29+ languages**.
|
| 32 |
+
|
| 33 |
+
### **Quickstart with Transformers**
|
| 34 |
+
|
| 35 |
+
```python
|
| 36 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 37 |
+
|
| 38 |
+
model_name = "prithivMLmods/Viper-OneCoder-UIGEN"
|
| 39 |
+
|
| 40 |
+
model = AutoModelForCausalLM.from_pretrained(
|
| 41 |
+
model_name,
|
| 42 |
+
torch_dtype="auto",
|
| 43 |
+
device_map="auto",
|
| 44 |
+
trust_remote_code=True
|
| 45 |
+
)
|
| 46 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 47 |
+
|
| 48 |
+
prompt = "Create a responsive navigation bar using Tailwind CSS."
|
| 49 |
+
messages = [
|
| 50 |
+
{"role": "system", "content": "You are an advanced AI assistant with expert-level UI coding and reasoning abilities."},
|
| 51 |
+
{"role": "user", "content": prompt}
|
| 52 |
+
]
|
| 53 |
+
text = tokenizer.apply_chat_template(
|
| 54 |
+
messages,
|
| 55 |
+
tokenize=False,
|
| 56 |
+
add_generation_prompt=True
|
| 57 |
+
)
|
| 58 |
+
model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
|
| 59 |
+
|
| 60 |
+
generated_ids = model.generate(
|
| 61 |
+
**model_inputs,
|
| 62 |
+
max_new_tokens=512
|
| 63 |
+
)
|
| 64 |
+
generated_ids = [
|
| 65 |
+
output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
|
| 66 |
+
]
|
| 67 |
+
|
| 68 |
+
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
| 69 |
+
print(response)
|
| 70 |
+
```
|
| 71 |
+
|
| 72 |
+
### **Intended Use**
|
| 73 |
+
- **Elite Web Development & UI Design**: Best-in-class model for writing, analyzing, and optimizing front-end code.
|
| 74 |
+
- **Step-by-Step Coding Logic Breakdown**: Guides developers through structured programming approaches and best practices.
|
| 75 |
+
- **Component-Based UI Development**: Generates reusable Tailwind and React components with clear explanations.
|
| 76 |
+
- **Structured Data Processing**: Handles JSON, XML, and structured UI component automation.
|
| 77 |
+
- **Multilingual Programming Support**: Proficient in HTML, CSS, Tailwind, JavaScript, React, Python, and Go.
|
| 78 |
+
- **Extended Technical Content Generation**: Ideal for writing documentation, blog posts, and front-end tutorials.
|
| 79 |
+
|
| 80 |
+
### **Limitations**
|
| 81 |
+
1. **High Computational Demand**: Requires powerful GPUs/TPUs for smooth inference due to **14B parameters**.
|
| 82 |
+
2. **Framework-Specific Variability**: Performance may vary across different front-end frameworks.
|
| 83 |
+
3. **Possible Error Propagation**: Extended text outputs might introduce logical inconsistencies.
|
| 84 |
+
4. **Limited Real-World Awareness**: The model does not have access to real-time internet updates.
|
| 85 |
+
5. **Prompt Sensitivity**: Performance depends on how well the prompt is structured.
|