From 17c430042e702b9d64eeed57f062bc039c16c29a Mon Sep 17 00:00:00 2001
From: lb203 <62638829+LinB203@users.noreply.github.com>
Date: Thu, 14 Mar 2024 13:21:23 +0800
Subject: [PATCH] Update sample.sh

Former-commit-id: b149a300f1aff53d9ab6e66a91cc2bcd37900347
---
 scripts/sample.sh | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/scripts/sample.sh b/scripts/sample.sh
index bf0e7f49d..fe5093daf 100644
--- a/scripts/sample.sh
+++ b/scripts/sample.sh
@@ -1,6 +1,15 @@
-accelerate launch --num_processes 1 --main_process_port 29501 --mixed_precision bf16 opensora/sample/sample.py \
+accelerate launch \
+  --num_processes 1 \
+  --main_process_port 29501 \
+  opensora/sample/sample.py \
   --model Latte-XL/122 \
   --ae stabilityai/sd-vae-ft-mse \
-  --ckpt results/011-Latte-XL-122-F128S3-landscope_feature-Gc-Amp/checkpoints/0005000.pt --extras 1 \
-  --fps 10 --num-frames 128 --image-size 256 --num-sampling-steps 250 --attention-mode flash
+  --ckpt 0050000.pt \
+  --extras 1 \
+  --fps 10 \
+  --num-frames 128 \
+  --image-size 256 \
+  --num-sampling-steps 250 \
+  --attention-mode flash \
+  -mixed_precision bf16