Update generate_story.sh

This commit is contained in:
Norbert Schmidt
2026-03-31 19:16:17 +02:00
parent 02b8c27835
commit b7b17a7290

View File

@@ -28,11 +28,13 @@ if [ -z "$_MLX_BG" ] && [ -t 0 ]; then
fi fi
# Default settings # Default settings
WIDTH=1920 WIDTH=1280
HEIGHT=1088 HEIGHT=768
FRAMES=121 FRAMES=121
STRENGTH=0.7 STRENGTH=0.7
FPS=24 FPS=24
PIPELINE="dev-two-stage-hq"
MODEL_REPO="prince-canuma/LTX-2.3-dev"
VENV_PYTHON="${VENV_PYTHON:-./venv/bin/python}" VENV_PYTHON="${VENV_PYTHON:-./venv/bin/python}"
OUTPUT_DIR="$HOME/Nextcloud/Documents/mlx-video-stories" OUTPUT_DIR="$HOME/Nextcloud/Documents/mlx-video-stories"
@@ -85,6 +87,8 @@ while [ $# -gt 0 ]; do
--frames) FRAMES="$2"; shift 2 ;; --frames) FRAMES="$2"; shift 2 ;;
--strength) STRENGTH="$2"; shift 2 ;; --strength) STRENGTH="$2"; shift 2 ;;
--fps) FPS="$2"; shift 2 ;; --fps) FPS="$2"; shift 2 ;;
--pipeline) PIPELINE="$2"; shift 2 ;;
--model-repo) MODEL_REPO="$2"; shift 2 ;;
--python) VENV_PYTHON="$2"; shift 2 ;; --python) VENV_PYTHON="$2"; shift 2 ;;
*) echo "Unknown option: $1"; usage ;; *) echo "Unknown option: $1"; usage ;;
esac esac
@@ -158,7 +162,8 @@ for i in $(seq 1 $NUM_SCENES); do
# First scene: Text-to-Video # First scene: Text-to-Video
$VENV_PYTHON -m mlx_video.models.ltx_2.generate \ $VENV_PYTHON -m mlx_video.models.ltx_2.generate \
--prompt "$PROMPT" \ --prompt "$PROMPT" \
--model-repo prince-canuma/LTX-2.3-distilled \ --pipeline $PIPELINE \
--model-repo $MODEL_REPO \
--text-encoder-repo google/gemma-3-12b-it \ --text-encoder-repo google/gemma-3-12b-it \
--height $HEIGHT \ --height $HEIGHT \
--width $WIDTH \ --width $WIDTH \
@@ -187,7 +192,8 @@ for i in $(seq 1 $NUM_SCENES); do
# Generate with I2V # Generate with I2V
$VENV_PYTHON -m mlx_video.models.ltx_2.generate \ $VENV_PYTHON -m mlx_video.models.ltx_2.generate \
--prompt "$PROMPT" \ --prompt "$PROMPT" \
--model-repo prince-canuma/LTX-2.3-distilled \ --pipeline $PIPELINE \
--model-repo $MODEL_REPO \
--text-encoder-repo google/gemma-3-12b-it \ --text-encoder-repo google/gemma-3-12b-it \
--image "$LAST_FRAME" \ --image "$LAST_FRAME" \
--image-strength $STRENGTH \ --image-strength $STRENGTH \