Views
No views yet
transformers and vllm libraries.
1from pprint import pprint
2from datasets import Dataset
3from vllm import SamplingParams
4from transformers import set_seed
5from bonito import Bonito
6
7unannotated_paragraph = """灌区以往的闸门控制系统在实际应用过程中普遍以人工操作为主,容易受到多种因素的影响,不可避免出现较多缺陷。如操作人员自身的综合能力、业务水平、工作态度等对工作质量和效率产生较大影响;工作人员实践操作中遇到极端气候、工作环境恶劣等问题,大大增加了工作难度,并存在较多安全隐患。"""
8pprint(unannotated_paragraph)
9bonito = Bonito("kitsdk/bonito-chinese-v1")
10set_seed(2)
11
12def convert_to_dataset(text):
13 dataset = Dataset.from_list([{"input": text}])
14 return dataset
15
16sampling_params = SamplingParams(max_tokens=256, top_p=0.95, temperature=0.5, n=1)
17synthetic_dataset = bonito.generate_tasks(
18 convert_to_dataset(unannotated_paragraph),
19 context_col="input",
20 task_type="mcqa",
21 sampling_params=sampling_params
22)
23pprint("----Generated Instructions----")
24pprint(f'Input: {synthetic_dataset[0]["input"]}')
25pprint(f'Output: {synthetic_dataset[0]["output"]}')