1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "smirki/UIGEN-T1.5"
4tokenizer = AutoTokenizer.from_pretrained(model_name)
5model = AutoModelForCausalLM.from_pretrained(model_name).to("cuda")
6
7prompt = """<|im_start|>user
8Design a sleek, modern dashboard for monitoring solar panel efficiency.<|im_end|>
9<|im_start|>assistant
10<|im_start|>think
11"""
12
13inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
14outputs = model.generate(**inputs, max_new_tokens=12012, do_sample=True, temperature=0.7)
15
16print(tokenizer.decode(outputs[0], skip_special_tokens=True))
1@misc{Tesslate_UIGEN-T1.5,
2 title={UIGEN-T1.5: Advanced Chain-of-Thought UI Generation Model},
3 author={smirki},
4 year={2025},
5 publisher={Hugging Face},
6 url={https://huggingface.co/Tesslate/UIGEN-T1.5}
7}