Views
No views yet
trust_remote_code=True because it uses a custom recurrence-based architecture.1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3tokenizer = AutoTokenizer.from_pretrained("Danilosistu/open-mythos-140m")
4model = AutoModelForCausalLM.from_pretrained(
5"Danilosistu/open-mythos-140m",
6trust_remote_code=True
7)
8
9inputs = tokenizer("Once upon a time", return_tensors="pt")
10outputs = model.generate(**inputs, max_new_tokens=50)
11print(tokenizer.decode(outputs[0]))