Views
No views yet
1message = [
2 {"role": "system", "content": "You are a helpful assistant chatbot."},
3 {"role": "user", "content": "두 개의 구가 각각 지름이 1, 2일때 두 구의 부피는 몇배지? 설명도 같이 해줘."}
4]
5tokenizer = AutoTokenizer.from_pretrained(hf_model)
6prompt = tokenizer.apply_chat_template(message, add_generation_prompt=True, tokenize=False)
7
8pipeline = transformers.pipeline(
9 "text-generation",
10 model=hf_model,
11 tokenizer=tokenizer
12)
13
14
15sequences = pipeline(
16 prompt,
17 do_sample=True,
18 temperature=0.7,
19 top_p=0.9,
20 num_return_sequences=1,
21 max_length=512,
22)
23print(sequences[0]['generated_text'])