mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 17:24:32 +02:00
prompt enhance nsfw allow/disallow
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -146,6 +146,7 @@ class APIProcess():
|
||||
prompt=req.prompt,
|
||||
system=req.system_prompt,
|
||||
seed=seed,
|
||||
nsfw=req.nsfw,
|
||||
)
|
||||
elif req.type == 'image':
|
||||
from modules.scripts import scripts_txt2img
|
||||
@@ -157,6 +158,7 @@ class APIProcess():
|
||||
system=req.system_prompt,
|
||||
image=decode_base64_to_image(req.image),
|
||||
seed=seed,
|
||||
nsfw=req.nsfw,
|
||||
)
|
||||
elif req.type == 'video':
|
||||
from modules.ui_video_vlm import enhance_prompt
|
||||
@@ -167,6 +169,7 @@ class APIProcess():
|
||||
prompt=req.prompt,
|
||||
model=model,
|
||||
system_prompt=req.system_prompt,
|
||||
nsfw=req.nsfw,
|
||||
)
|
||||
else:
|
||||
raise HTTPException(status_code=400, detail="prompt enhancement: invalid type")
|
||||
|
||||
Reference in New Issue
Block a user