CoolFace
Apppublic

MacLeanLuke/gemma-2b-tool-tuned-code

sourceHugging Facemitupdated 2y agoView on Hugging Face
0likes
run.py10 linesDownload Raw Back to root
1from transformers import AutoModelForCausalLM, AutoTokenizer2 3repo_id = "MacLeanLuke/gemma-2b-tool-tuned"4 5model = AutoModelForCausalLM.from_pretrained(repo_id)6tokenizer = AutoTokenizer.from_pretrained(repo_id)7 8inputs = tokenizer("Hello, how are you?", return_tensors="pt")9outputs = model.generate(**inputs)10print(tokenizer.decode(outputs[0]))