Views
No views yet
SmolLM-135M-Instruct-layer-width-pruned-90M-rawHuggingFaceTB/SmolLM-135M-Instruct1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "your-username/SmolLM-135M-Instruct-layer-width-pruned-90M-raw"
4tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/SmolLM-135M-Instruct")
5model = AutoModelForCausalLM.from_pretrained(model_name)
6
7inputs = tokenizer("Hello world!", return_tensors="pt")
8outputs = model.generate(**inputs, max_length=50)
9print(tokenizer.decode(outputs[0]))