mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
add experimental openai interface
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -375,6 +375,7 @@ class ReqPromptEnhance(BaseModel):
|
||||
process_words: Optional[str] = Field(title="Banned words", default=None, description="List of words to process")
|
||||
semantic_threshold: Optional[float] = Field(title="Semantic threshold", default=None, description="Semantic similarity threshold for processed words")
|
||||
embedding_similarity: Optional[float] = Field(title="Embedding similarity", default=None, description="Embedding similarity threshold for processed words")
|
||||
use_openai: Optional[bool] = Field(title="Use OpenAI", default=False, description="Use OpenAI API for model access")
|
||||
|
||||
class ResPromptEnhance(BaseModel):
|
||||
prompt: str = Field(title="Prompt", description="Enhanced prompt")
|
||||
|
||||
@@ -247,6 +247,7 @@ class APIProcess:
|
||||
process_words=req.process_words,
|
||||
semantic_threshold=req.semantic_threshold,
|
||||
embedding_similarity=req.embedding_similarity,
|
||||
use_openai=req.use_openai
|
||||
)
|
||||
elif req.type == 'video':
|
||||
from modules.ui_video_vlm import enhance_prompt
|
||||
|
||||
Reference in New Issue
Block a user