Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained("a-k-aAiMGoD/phi3-mini-distributed-fine-tune")
4model = AutoModelForCausalLM.from_pretrained("a-k-aAiMGoD/phi3-mini-distributed-fine-tune")
5
6# Example usage
7text = "Hello, how are you?"
8inputs = tokenizer(text, return_tensors="pt")
9outputs = model.generate(**inputs, max_length=100)
10response = tokenizer.decode(outputs[0], skip_special_tokens=True)
11print(response)