Views
No views yet
1from transformers import AutoConfig, AutoModel, AutoTokenizer
2import transformers
3import torch
4hf_model = 'mncai/mistral-7b-v5'
5message = "### Instruction:\n\n두 개의 구가 각각 지름이 1, 2일때 두 구의 부피는 몇배지? 설명도 같이 해줘.\n\n### Response:\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']}")