feat: support config.allowAnlas

This commit is contained in:
Shigma 2022-10-09 04:12:12 +08:00
parent 1c7a4bbff4
commit f7183fcf82
No known key found for this signature in database
GPG Key ID: 21C89B0B92907E14
3 changed files with 22 additions and 15 deletions

View File

@ -161,6 +161,13 @@ console.log(JSON.parse(localStorage.session).auth_token)
默认情况下是否过滤不良构图。
### allowAnlas
- 类型:`boolean`
- 默认值:`true`
是否允许使用点数。禁用后部分功能 (图片增强和手动设置某些参数) 将无法使用。
### basePrompt
- 类型: `string`

View File

@ -51,7 +51,7 @@ export const Config: Schema<Config> = Schema.object({
orient: Schema.union(orients).description('默认的图片方向。').default('portrait'),
sampler: Schema.union(samplers).description('默认的采样器。').default('k_euler_ancestral'),
anatomy: Schema.boolean().default(true).description('是否过滤不合理构图。'),
allowAnlas: Schema.boolean().default(true).description('是否允许使用点数。'),
allowAnlas: Schema.boolean().default(true).description('是否允许使用点数。禁用后部分功能 (图片增强和手动设置某些参数) 将无法使用。'),
basePrompt: Schema.string().description('默认的附加标签。').default('masterpiece, best quality'),
forbidden: Schema.string().role('textarea').description('违禁词列表。含有违禁词的请求将被拒绝。').default(''),
endpoint: Schema.string().description('API 服务器地址。').default('https://api.novelai.net'),
@ -106,19 +106,21 @@ export function apply(ctx: Context, config: Config) {
if (!input?.trim()) return session.execute('help novelai')
let imgUrl: string
input = segment.transform(input, {
image(attrs) {
imgUrl = attrs.url
return ''
},
})
if (config.allowAnlas) {
input = segment.transform(input, {
image(attrs) {
imgUrl = attrs.url
return ''
},
})
if (options.enhance && !imgUrl) {
return session.text('.expect-image')
}
if (options.enhance && !imgUrl) {
return session.text('.expect-image')
}
if (!input.trim() && !config.basePrompt) {
return session.text('.expect-prompt')
if (!input.trim() && !config.basePrompt) {
return session.text('.expect-prompt')
}
}
input = input.toLowerCase().replace(/[,]/g, ', ').replace(/\s+/g, ' ')

View File

@ -13,15 +13,13 @@ commands:
sampler: 设置采样器 (k_euler_ancestral, k_euler, k_lms, plms, ddim)
anatomy.true: 过滤不合理构图
anatomy.false: 允许不合理构图
anatomy=true: 过滤不合理构图
anatomy=false: 允许不合理构图
seed: 设置随机种子
messages:
expect-prompt: 请输入标签。
expect-image: 请输入图片。
invalid-input: 只接受英文输入。
forbidden-word: 请求含有违禁词 ({0})
forbidden-word: 输入含有违禁词
concurrent-jobs: 请稍后再试。
waiting: 在画了在画了
unknown-error: 发生未知错误。