let5see/AI-Video
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

支持runninghub并发限制可配置

Browse Source
This commit is contained in:
puke
2025-12-28 14:56:22 +08:00
parent 5a2334ccde
commit 1530d1f8c8
7 changed files with 29 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pixelle_video/config/manager.py
View File

@@ -130,6 +130,7 @@ class ConfigManager:
"comfyui_url": self.config.comfyui.comfyui_url,
"comfyui_api_key": self.config.comfyui.comfyui_api_key,
"runninghub_api_key": self.config.comfyui.runninghub_api_key,
"runninghub_concurrent_limit": self.config.comfyui.runninghub_concurrent_limit,
"tts": {
"default_workflow": self.config.comfyui.tts.default_workflow,
},
@@ -147,7 +148,8 @@ class ConfigManager:
self,
comfyui_url: Optional[str] = None,
comfyui_api_key: Optional[str] = None,
runninghub_api_key: Optional[str] = None
runninghub_api_key: Optional[str] = None,
runninghub_concurrent_limit: Optional[int] = None
):
"""Set ComfyUI global configuration"""
updates = {}
@@ -157,6 +159,8 @@ class ConfigManager:
updates["comfyui_api_key"] = comfyui_api_key
if runninghub_api_key is not None:
updates["runninghub_api_key"] = runninghub_api_key
if runninghub_concurrent_limit is not None:
updates["runninghub_concurrent_limit"] = runninghub_concurrent_limit
if updates:
self.update({"comfyui": updates})

1
pixelle_video/config/schema.py
View File

@@ -73,6 +73,7 @@ class ComfyUIConfig(BaseModel):
comfyui_url: str = Field(default="http://127.0.0.1:8188", description="ComfyUI Server URL")
comfyui_api_key: Optional[str] = Field(default=None, description="ComfyUI API Key (optional)")
runninghub_api_key: Optional[str] = Field(default=None, description="RunningHub API Key (optional)")
runninghub_concurrent_limit: int = Field(default=1, ge=1, le=10, description="RunningHub concurrent execution limit (1-10)")
tts: TTSSubConfig = Field(default_factory=TTSSubConfig, description="TTS-specific configuration")
image: ImageSubConfig = Field(default_factory=ImageSubConfig, description="Image-specific configuration")
video: VideoSubConfig = Field(default_factory=VideoSubConfig, description="Video-specific configuration")