使用vllm部署
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

41 lines
2.9 KiB

import math
class Config:
def __init__(self):
# 目录提取拼接相关参数
self.pantten_second_biaoti = '[2二ⅡⅠ][、.]\s{0,}?[\u4e00-\u9fa5]+'
self.pantten_other_biaoti = '[2-9二三四五六七八九ⅡⅢⅣⅤⅥⅦⅧⅨ][、.]\s{0,}?[\u4e00-\u9fa5]+'
self.pantten_biaoti = '[1-9一二三四五六七八九ⅠⅡⅢⅣⅤⅥⅦⅧⅨ][、.]\s{0,}?[\u4e00-\u9fa5a-zA-Z]+'
# chatgpt 接口相关参数
self.mulu_prompt = "生成目录#\n问:为论文题目《{}》生成目录,要求只有一级标题和二级标题,一级标题使用中文数字 例如一、xxx;二级标题使用阿拉伯数字 例如1.1 xxx;一级标题不少于7个;每个一级标题至少包含3个二级标题\n答:\n"
self.first_title_prompt = "生成论文小标题内容#\n问:论文题目是《{}》,目录是“{}”,请把其中的大标题“{}”的内容补充完整,补充内容字数在{}字左右\n答:\n"
self.small_title_prompt = "生成论文小标题内容#\n问:论文题目是《{}》,目录是“{}”,请把其中的小标题“{}”的内容补充完整,补充内容字数在{}字左右\n答:\n"
self.references_prompt = "论文题目是“{}”,目录是“{}”,请为这篇论文生成15篇左右的参考文献,要求其中有有中文参考文献不低于12篇,英文参考文献不低于2篇"
self.thank_prompt = "请以“{}”为题写一篇论文的致谢"
self.kaitibaogao_prompt = "请以《{}》为题目生成研究的主要的内容、背景、目的、意义,要求不少于1500字"
self.chinese_abstract_prompt = "生成论文摘要#\n问:论文题目是《{}》,目录是“{}”,生成论文摘要,要求生成的字数在600字左右\n答:\n"
self.english_abstract_prompt = "翻译摘要#\n问:请把“{}”这段文字翻译成英文\n答:\n"
self.chinese_keyword_prompt = "生成关键字#\n问:请为“{}”这段论文摘要生成3-5个关键字,使用阿拉伯数字作为序号标注,例如“1.xxx \n2.xxx \n3.xxx \n4.xxx \n5.xxx \n\"\n答:\n"
self.english_keyword_prompt = "翻译关键词#\n问:请把“{}”这几个关键字翻译成英文\n答:\n"
self.dabiaoti = ["", "", "", "", "", "", "", ""]
self.project_data_txt_path = "/home/majiahui/project2/LLaMA-Efficient-Tuning/new_data_txt_4"
# 流程相关参数
self.thanks = "致谢"
self.references = "参考文献"
self.excursus = "附录"
self.u = 3.5 # 均值μ
self.sig = math.sqrt(6.0)
self.zong_gradient = 6
self.paper_word_count = 12000
# flask port
self.flask_port = "14003"
# redis config
self.reids_ip = '192.168.31.145'
self.reids_port = 6379
self.reids_db = 7
self.reids_password = 'Zhicheng123*'