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.
|