Views
No views yet
GPT2LMHeadModel for compatibility with the wider ecosystem..gguf build? See mondk/GGUF.msh-tiny.<|user|>
{your message}
<|assistant|><|end|>.1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model = AutoModelForCausalLM.from_pretrained("mondk/Msh-Tiny-14M")
5tokenizer = AutoTokenizer.from_pretrained("mondk/Msh-Tiny-14M")
6
7prompt = "<|user|>\nhi\n<|assistant|>\n"
8input_ids = tokenizer(prompt, return_tensors="pt").input_ids
9output = model.generate(input_ids, max_new_tokens=100, do_sample=True, temperature=0.7, top_k=40)
10print(tokenizer.decode(output[0][input_ids.shape[1]:], skip_special_tokens=True))tatsu-lab/alpacateknium/OpenHermes-2.5HuggingFaceH4/no_robots