storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-18 00:33:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update preview_server.py

Browse Source
This commit is contained in:
salt
2024-03-28 09:26:08 +00:00
committed by GitHub
parent c3f49d9a9d
commit 940616a817
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
workflows/preview_server/preview_server.py
+2 -1
View File
@@ -28,6 +28,7 @@ COMFY_ROOT_DIRECTORY = os.environ.get("COMFY_ROOT_DIRECTORY",
"/home/salt/clone/storyteller/storyteller-ml/workflows/comfy/ComfyUI")
SERVER_START_TIMEOUT = int(os.environ.get("SERVER_START_TIMEOUT", 60))
STYLE_LOCK = os.environ.get("STYLE_LOCK", None)
PREVIEW_WORKFLOW_NAME = os.environ.get("PREVIEW_WORKFLOW_NAME", None)
style_to_filename = {
"anime_2_5d": "1_2.5d_anime_model.json",
@@ -117,7 +118,7 @@ async def process_video(video_path: str, style_filename, positive_prompt=None, n
mappings_directory = Path(V2V_WORKFLOWS_DIRECTORY) / "mappings"
style_json = json.load(open(styles_directory / style_filename))
workflow_json = json.load(open(workflow_directory / style_json["workflow_api_name"]))
workflow_json = json.load(open(workflow_directory / PREVIEW_WORKFLOW_NAME))
mapping_json = json.load(open(mappings_directory / style_json["mapping_name"]))
json_mods = style_json["modifications"]