Views
No views yet


1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4torch.set_default_device("cuda")
5
6model = AutoModelForCausalLM.from_pretrained("abideen/Mistral-v0.2-orpo", torch_dtype="auto", trust_remote_code=True)
7tokenizer = AutoTokenizer.from_pretrained("abideen/Mistral-v0.2-orpo", trust_remote_code=True)
8
9inputs = tokenizer('''
10 """
11 Write a detailed analogy between mathematics and a lighthouse.
12 """''', return_tensors="pt", return_attention_mask=False)
13
14outputs = model.generate(**inputs, max_length=200)
15text = tokenizer.batch_decode(outputs)[0]
16print(text)
17