updated venv path, image inputs naming corrected

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
@@ -162,19 +162,13 @@ def main():
input_media_filename = args.portrait_media_filename
input_driver_filename = args.driver_media_filename
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):
ensure_server_is_healthy(SERVER_IP, restart_trigger_file)
ensure_input_directory_empty()
ensure_output_directory_empty()
input_filename = "input.mp4"
if input_media_filename:
if input_is_image:
@@ -185,7 +179,13 @@ def main():
if input_driver_filename:
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()
print("running pipeline" + str(pipeline_type))
enqueue_prompt_and_wait(prompt)
@@ -239,7 +239,6 @@ def enqueue_prompt_and_wait(prompt):
time.sleep(1)
def generate_prompt_for_style(
workflow_filename: str,
input_media_filename: Optional[str] = None,
@@ -3,7 +3,7 @@
set -ex
cd /app/venv/bin/activate
source /app/venv/bin/activate
date > /restart.txt