Views
No views yet
1# Load model directly
2from transformers import AutoTokenizer, AutoModelForCausalLM
3
4tokenizer = AutoTokenizer.from_pretrained("kviai/KviGPT-7b-Chat")
5model = AutoModelForCausalLM.from_pretrained("kviai/KviGPT-7b-Chat")
6
7prompt = "Hi, what do you know about TON coin?"
8output = pipeline(prompt)
9print(output)