Views
No views yet
sample-10BT for achieving good results with low token size.1from transformers import AutoModelForCausalLM, AutoTokenizer
2tokenizer = AutoTokenizer.from_pretrained("merterbak/Seed-0.5B")
3model = AutoModelForCausalLM.from_pretrained(
4 "merterbak/Seed-0.5B",
5 trust_remote_code=True,
6 dtype="auto"
7)
8prompt = "Climate change can affect"
9inputs = tokenizer(prompt, return_tensors="pt")
10outputs = model.generate(
11 **inputs,
12 temperature=0.3,
13 top_k=40,
14 do_sample=True,
15 repetition_penalty=1.2,
16 no_repeat_ngram_size=3,
17)
18print(tokenizer.decode(outputs[0], skip_special_tokens=True))