Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3import torch
4
5base = AutoModelForCausalLM.from_pretrained(
6 "Qwen/Qwen2.5-Coder-7B-Instruct",
7 torch_dtype=torch.bfloat16,
8 device_map="auto"
9)
10tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-Coder-7B-Instruct")
11model = PeftModel.from_pretrained(base, "lhordking/webcoder-7b")
12
13prompt = "Create a responsive dark mode landing page for a SaaS product"
14inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
15output = model.generate(**inputs, max_new_tokens=1024)
16print(tokenizer.decode(output[0], skip_special_tokens=True))"Create a responsive navbar with dark mode toggle""Build a SaaS landing page with hero section and pricing table""Make a login form with email and password validation""Create a portfolio page with project cards and animations"