mirror of
https://github.com/storytold/artcraft.git
synced 2026-10-20 17:02:53 -04:00
54 lines
1.8 KiB
Bash
Executable File
54 lines
1.8 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# This script is to run live portrait on bare metal local dev.
|
|
# It's probably preferable to run on local k8s, but this can be used in the interim.
|
|
# TODO(bt,2024-08-18): Remove hardcoded paths
|
|
|
|
set -euxo pipefail
|
|
|
|
RUST_PROJECT_ROOT="${HOME}/dev/storyteller/storyteller-rust"
|
|
ML_PROJECT_ROOT="${HOME}/dev/storyteller/storyteller-ml"
|
|
|
|
export SCOPED_EXECUTION_JOB_TYPES=live_portrait
|
|
|
|
export COMFY_INFERENCE_ROOT_DIRECTORY="${ML_PROJECT_ROOT}/workflows/comfy"
|
|
|
|
export COMFY_INFERENCE_EXECUTABLE_OR_COMMAND="venv/bin/python ./ComfyLauncher/ComfyLivePortraitRunner.py --server-restart-trigger-file /tmp/restart.txt"
|
|
|
|
export COMFY_LAUNCH_COMMAND="bash ${ML_PROJECT_ROOT}/workflows/comfy/live-runner-comfy-server-starter.sh"
|
|
|
|
export BASE_LIVE_PORTRAIT_WORKFLOW="25-07-2024/yae_LivingPortrait(Cuda)_25-07_API.json"
|
|
|
|
# For downstream Python script
|
|
export V2V_WORKFLOWS_DIRECTORY="${ML_PROJECT_ROOT}/workflows/comfy/workflow_configs"
|
|
export COMFY_ROOT_DIRECTORY="${ML_PROJECT_ROOT}/workflows/comfy/ComfyUI"
|
|
|
|
# Watermarks
|
|
export FAKEYOU_WATERMARK_PATH="${RUST_PROJECT_ROOT}/includes/container_includes/image_assets/fakeyou_watermark.png"
|
|
export STORYTELLER_WATERMARK_PATH="${RUST_PROJECT_ROOT}/includes/container_includes/image_assets/storyteller_watermark.png"
|
|
|
|
export FAKEYOU_WATERMARK_SCALE="0.2"
|
|
|
|
cargo run --bin inference-job
|
|
|
|
# Other args:
|
|
|
|
# MAIN_IPA_WORKFLOW
|
|
# FACE_DETAILER_WORKFLOW
|
|
# UPSCALER_WORKFLOW
|
|
|
|
# COMFY_INFERENCE_CONFIG_PATH
|
|
# COMFY_INFERENCE_EXECUTABLE_OR_COMMAND
|
|
# COMFY_SETUP_SCRIPT
|
|
# COMFY_LAUNCH_COMMAND
|
|
# COMFY_STARTUP_HEALTHCHECK_ENABLED
|
|
# COMFY_INFERENCE_MAYBE_VENV_COMMAND
|
|
# COMFY_TIMEOUT_SECONDS
|
|
# COMFY_INFERENCE_MAYBE_DOCKER_IMAGE
|
|
# COMFY_MOUNTS_DIRECTORY
|
|
# COMFY_VIDEO_PROCESSING_SCRIPT
|
|
# COMFY_STYLES_DIRECTORY
|
|
# COMFY_WORKFLOWS_DIRECTORY
|
|
# COMFY_MAPPINGS_DIRECTORY
|
|
|