Views
No views yet
1from transformers import AutoConfig, AutoModel, AutoTokenizer
2import transformers
3import torch
4hf_model = 'mncai/mistral-7b-dpo-v5'
5message = "<|user|>\n두 개의 구가 있는데 각각 지름이 1, 2일때 각 구의 부피는 몇배야? 설명도 같이 해줘.\n<|assistant|>\n"
6
7sequences = pipeline(
8 message,
9 do_sample=True,
10 top_k=10,
11 num_return_sequences=1,
12 eos_token_id=tokenizer.eos_token_id,
13 max_length=2048,
14)
15for seq in sequences:
16 print(f"Result: {seq['generated_text']}")