Views
No views yet
Qwen3ForCausalLM).1# transformers (completion)
2from transformers import AutoModelForCausalLM, AutoTokenizer
3repo = "manueldeprada/littlelearner-5b-bounded-base"
4tok = AutoTokenizer.from_pretrained(repo)
5model = AutoModelForCausalLM.from_pretrained(repo, dtype="bfloat16", device_map="cuda")
6ids = tok("The sum of 2 and 3 is", return_tensors="pt").to(model.device)
7print(tok.decode(model.generate(**ids)[0], skip_special_tokens=True))1# vLLM
2from vllm import LLM
3llm = LLM("manueldeprada/littlelearner-5b-bounded-base")
4print(llm.generate(["The sum of 2 and 3 is"])[0].outputs[0].text)