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

新增runninghub的48G显存机器调用支持

Browse Source
This commit is contained in:
puke
2026-01-06 11:11:11 +08:00
parent 7d78dcd078
commit d6b2f02cce
9 changed files with 69 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pixelle_video/config/manager.py
View File

@@ -131,6 +131,7 @@ class ConfigManager:
"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,
"runninghub_instance_type": self.config.comfyui.runninghub_instance_type,
"tts": {
"default_workflow": self.config.comfyui.tts.default_workflow,
},
@@ -149,7 +150,8 @@ class ConfigManager:
comfyui_url: Optional[str] = None,
comfyui_api_key: Optional[str] = None,
runninghub_api_key: Optional[str] = None,
runninghub_concurrent_limit: Optional[int] = None
runninghub_concurrent_limit: Optional[int] = None,
runninghub_instance_type: Optional[str] = None
):
"""Set ComfyUI global configuration"""
updates = {}
@@ -161,6 +163,9 @@ class ConfigManager:
updates["runninghub_api_key"] = runninghub_api_key
if runninghub_concurrent_limit is not None:
updates["runninghub_concurrent_limit"] = runninghub_concurrent_limit
if runninghub_instance_type is not None:
# Empty string means disable (treat as None for storage)
updates["runninghub_instance_type"] = runninghub_instance_type if runninghub_instance_type else None
if updates:
self.update({"comfyui": updates})