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

updated venv path, image inputs naming corrected

Browse Source
This commit is contained in:
Kasisnu
2024-07-26 16:44:56 +05:30
parent 2564584d63
commit 42d712ff4c
2 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
workflows/comfy/ComfyLauncher/ComfyLivePortraitRunner.py
+9 -10
View File
@@ -162,19 +162,13 @@ def main():
input_media_filename = args.portrait_media_filename input_media_filename = args.portrait_media_filename
input_driver_filename = args.driver_media_filename input_driver_filename = args.driver_media_filename
input_is_image = args.input_is_image input_is_image = args.input_is_image
input_filename = "input.mp4"
print("Generating prompt json using workflow: " + workflow_filename)
prompt = generate_prompt_for_style(
workflow_filename=workflow_filename,
input_media_filename="input/" + input_filename,
input_driver_filename="input/driver.mp4",
input_is_image=input_is_image,
)
for _ in range(outer_retries): for _ in range(outer_retries):
ensure_server_is_healthy(SERVER_IP, restart_trigger_file) ensure_server_is_healthy(SERVER_IP, restart_trigger_file)
ensure_input_directory_empty() ensure_input_directory_empty()
ensure_output_directory_empty() ensure_output_directory_empty()
input_filename = "input.mp4"
if input_media_filename: if input_media_filename:
if input_is_image: if input_is_image:
@@ -185,7 +179,13 @@ def main():
if input_driver_filename: if input_driver_filename:
ensure_pipeline_input_present(input_driver_filename, "driver.mp4") ensure_pipeline_input_present(input_driver_filename, "driver.mp4")
print("Generating prompt json using workflow: " + workflow_filename)
prompt = generate_prompt_for_style(
workflow_filename=workflow_filename,
input_media_filename="input/" + input_filename,
input_driver_filename="input/driver.mp4",
input_is_image=input_is_image,
)
time_before = time.perf_counter() time_before = time.perf_counter()
print("running pipeline" + str(pipeline_type)) print("running pipeline" + str(pipeline_type))
enqueue_prompt_and_wait(prompt) enqueue_prompt_and_wait(prompt)
@@ -239,7 +239,6 @@ def enqueue_prompt_and_wait(prompt):
time.sleep(1) time.sleep(1)
def generate_prompt_for_style( def generate_prompt_for_style(
workflow_filename: str, workflow_filename: str,
input_media_filename: Optional[str] = None, input_media_filename: Optional[str] = None,
workflows/comfy/live-runner-comfy-server-starter.sh
+1 -1
View File
@@ -3,7 +3,7 @@
set -ex set -ex
cd /app/venv/bin/activate source /app/venv/bin/activate
date > /restart.txt date > /restart.txt