Views
No views yet
model.safetensors weights.1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3model_id = "Sharjeelbaig/neurocoder"
4tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
5model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
6
7prompt = "Generate a landing page for marketing agency titled Velocity Landing"
8inputs = tokenizer(prompt, return_tensors="pt")
9outputs = model.generate(
10 **inputs,
11 max_new_tokens=220,
12 do_sample=False,
13 repetition_penalty=1.22,
14 no_repeat_ngram_size=6,
15 use_cache=True,
16)
17text = tokenizer.decode(outputs[0], skip_special_tokens=True)
18print(text.split("\nAssistant:", 1)[-1].strip())