Views
No views yet
transformers library:1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_path = "." # Or the repository ID after pushing to Hugging Face Hub
4tokenizer = AutoTokenizer.from_pretrained(model_path)
5model = AutoModelForCausalLM.from_pretrained(model_path)
6
7# Example inference
8input_text = "def fibonacci("
9input_ids = tokenizer(input_text, return_tensors="pt").input_ids
10output_ids = model.generate(input_ids, max_length=50)
11output_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
12print(output_text)