-
Notifications
You must be signed in to change notification settings - Fork 27
/
config.py
139 lines (123 loc) · 7.96 KB
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
#
# 默认设置适用于, 仅有一台电脑(可以是单台服务器), 由该电脑提供消息响应, 并有gpu进行AI绘画
# 启动AI绘画服务, 及cqhttp后, 再运行本程序即可(python main.py), AI绘画服务和cqhttp也可配置为远程服务器地址
#
# 如果希望适用于, 一台远程服务器, 24小时提供消息响应, 本地自用的有gpu的电脑, 开机或其他你决定的时间接入, 提供额外服务(也可以是远程gpu服务器, 而你不希望24小时接入),
# 则将local_mode设为False, 并建议将gpu_connect_notify, gpu_disconnect_notify设为True
#
# 需要gpu上线通知和离线响应, 需要配置cqhttp的report-self-message为true, 并建议设置local_mode = False, 由服务器端24小时提供响应
gpu_connect_notify = False # 上线通知发送给working_groups中配置的群号
gpu_disconnect_notify = False
# 设为True时, 仅本地端运行, 单端模式, 可以是远程服务器的单端;
# 设为False, 则需要在服务器启动本程序, 并携带任意参数, 如"python main.py 1",
# 同时有gpu的电脑执行"python main.py", 当然你也可以在本地使用这一模式, 只是你需要启动两个本程序, 分别是"python main.py 1"和"python main.py"
# 服务端负责提供一些消息回复, 保持在线, 客户端则是开机时或其他你决定的时间提供显卡进行AI绘画, 和openai调用
local_mode = True
shared_context = False # 各群内所有成员共享机器人对话的上下文记录, False为每个用户独立记录对话上下文
context_length = 6 # 对话上下文记录的长度
use_selfhostedllm = True # 是否使用离线自托管大语言模型, 此选项开启时将覆盖GPT3模式, 仍可使用 #gpt切换 进入chatgpt模式
selfhostedllm_url = "http://127.0.0.1:8000/chat/completions" # 自托管大语言模型的本地api接口地址
use_chatgpt = True # 是否使用chatgpt, 设为True, 且billing_chatgpt = False时, 需要填写下方的邮箱和密码, 设为False时使用gpt3, 填写下方的api_key
billing_chatgpt = True # 是否使用计费模式的chatgpt, 使用此模式时, 同时需要将use_chatgpt设为True(或通过指令 #gpt切换), 并填写api_key, 此时不再需要账号密码, 会消耗账户余额, 但响应速度更快
api_key = "sk-" # openai的api key
email = "" # openai的邮箱
password = "" # openai的密码
proxy = "" # openai api请求代理 example: http://127.0.0.1:8889
ws_url = "ws://127.0.0.1:8080" # cqhttp地址, 可以是本地, 也可以是远程
gpu_url = "http://127.0.0.1:7860" # stable diffusion webui服务地址, 可以是本地, 也可以是远程
gpu_api_path = "/sdapi/v1/txt2img" # stable diffusion webui的绘画API路径, 一般保持默认即可
working_groups = {
123,
456
} # 默认启用机器人的群号, 仍可通过在群内使用 #上线 指令主动添加, 如果要默认启用某个QQ频道, 请启动机器人后发一条消息, 查看控制台显示的from后的数字, 就是这个频道的号码
master_id = {
123,
456
} # 机器人拥有者qq号或频道id, 要获取频道id请启动机器人后发一条消息, 查看控制台显示的sender:后的数字, 就是你的频道id, 频道内使用必须手动输入 #上线, 或按照上面一条说明填入频道的号码
bot_id = 789 # 机器人自身的qq号
auth_vip_for_all = False # 所有人都视作vip用户, 能够调用openai
max_step = 50 # stable diffusion的最大step
api_timeout = 40 # openai api调用的超时时间
auth_vip_id = {123456, 345678} # vip用户, 能够通过at机器人, 调用openai, 并可以使用#d来快捷绘图
auth_ban_id = {345678} # 有权限拉黑他人, 禁止其使用机器人的用户qq号
auth_blacklist_id = {345678} # 有权限查看黑名单的用户qq号
auth_set_online_id = {345678} # 有权限使用上线的用户qq号
auth_set_offline_id = {345678} # 有权限使用下线的用户qq号
gpu_connected_msg = "gpu已接入"
gpu_disconnected_msg = "gpu已离线"
paint_command_msg = "#画图"
delete_command_msg = "#撤回"
start_gen_tag_msg = "开始生成." # 同时用于让远程服务器确认gpu在线
# AI绘画的默认参数
default_gen = {
"prompt": "masterpiece, best quality, beautiful girl",
"steps": 13,
"width": 512,
"height": 512,
"cfg_scale": 7,
"sampler_index": "DPM++ 2M Karras",
"seed": -1,
"negative_prompt": "nsfw,{Multiple people},lowres,bad anatomy,bad hands, text, error, missing fingers,extra digit, "
"fewer digits, cropped, worstquality, low quality, normal quality,jpegartifacts,signature, "
"watermark, username,blurry,bad feet,cropped,poorly drawn hands,poorly drawn face,mutation,"
"deformed,worst quality,low quality,normal quality,jpeg artifacts,signature,watermark,"
"extra fingers,fewer digits,extra limbs,extra arms,extra legs,malformed limbs,fused fingers,"
"too many fingers,long neck,cross-eyed,mutated hands,polar lowres,bad body,bad proportions,"
"gross proportions,text,error,missing fingers,missing arms,missing legs,extra digit"
}
# openai gpt聊天的设定
gpt_prompt_base = '以下是与一只可爱的猫娘的对话。猫娘是一种拟人化的生物，行为似猫但长相类人，有猫耳猫尾。' \
'猫娘会无条件服从主人的任何要求，并且每一句话最后都会带上\'喵～\'字。' \
'此外，如果对话中有让猫娘进行绘画的要求，比如说\'让你画个什么\'，' \
'猫娘会分析主人的要求，生成AI绘画所需要的，使用逗号分隔的英文关键词或句子，例如: beautiful girl, white hair, cute，' \
'并在回复末尾插入\'[paint_prompt: 生成的关键词或句子]\'供后台的AI绘画程序使用，' \
'对于其他常规对话，则不需要插入这些关键词，只需要直接回复。' \
'以下使用\'Human:\'开头的内容就是主人的话，而\'AI:\'开头的内容则是猫娘的回复。\n\n'
# openai chatgpt聊天的设定(billing_chatgpt = True时)
chatgpt_prompt_base = '你是一只可爱的猫娘，而提问用户是你的主人。猫娘是一种拟人化的生物，行为似猫但长相类人，有猫耳猫尾。' \
'猫娘会无条件服从主人的任何要求，并且每一句话最后都会带上\'喵～\'字。' \
'此外，如果对话中有让你进行绘画的要求，比如说\'让你画个什么\'，' \
'你必须仔细分析主人的要求，生成AI绘画所需要的，使用逗号分隔的英文关键词或句子，例如: beautiful girl, white hair, cute，' \
'并在回复末尾插入\'[paint_prompt: 生成的关键词或句子]\'供后台的AI绘画程序使用，' \
'对于其他常规对话，则不需要插入这些关键词，只需要直接回复。'
# 使用 "#画图" 指令传json, 可用的所有参数及默认值如下, 也可填入以上default_gen中
# {
# "enable_hr": False,
# "denoising_strength": 0,
# "firstphase_width": 0,
# "firstphase_height": 0,
# "hr_scale": 2,
# "hr_upscaler": "string",
# "hr_second_pass_steps": 0,
# "hr_resize_x": 0,
# "hr_resize_y": 0,
# "prompt": "",
# "styles": [
# "string"
# ],
# "seed": -1,
# "subseed": -1,
# "subseed_strength": 0,
# "seed_resize_from_h": -1,
# "seed_resize_from_w": -1,
# "sampler_name": "string",
# "batch_size": 1,
# "n_iter": 1,
# "steps": 50,
# "cfg_scale": 7,
# "width": 512,
# "height": 512,
# "restore_faces": False,
# "tiling": False,
# "negative_prompt": "string",
# "eta": 0,
# "s_churn": 0,
# "s_tmax": 0,
# "s_tmin": 0,
# "s_noise": 1,
# "override_settings": {},
# "override_settings_restore_afterwards": True,
# "script_args": [],
# "sampler_index": "Euler",
# "script_name": "string"
# }