[fix] fix the path typo for google/mt5-xxl in gradio_web_server.py (#405

) * [fix] fix the path typo for google/mt5-xxl in gradio_web_server.py * [fix] fix the issue: the cache of pretrained mt5-xxl weights is inconsistent. --------- Co-authored-by: guangyi <[email protected]>
PKU-YuanGroup · Aug 23, 2024 · 294993c · 294993c
1 parent de7fc31
commit 294993c
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/opensora/serve/gradio_web_server.py b/opensora/serve/gradio_web_server.py
@@ -80,9 +80,9 @@ def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str
     # text_encoder = MT5EncoderModel.from_pretrained(args.text_encoder_name, cache_dir=args.cache_dir, 
     #                                                low_cpu_mem_usage=True, torch_dtype=weight_dtype)
     # tokenizer = AutoTokenizer.from_pretrained(args.text_encoder_name, cache_dir=args.cache_dir)
-    text_encoder = MT5EncoderModel.from_pretrained("/storage/ongoing/new/Open-Sora-Plan/cache_dir/mt5-xxl", cache_dir=args.cache_dir, 
+    text_encoder = MT5EncoderModel.from_pretrained("google/mt5-xxl", cache_dir=args.cache_dir, 
                                                    low_cpu_mem_usage=True, torch_dtype=weight_dtype)
-    tokenizer = AutoTokenizer.from_pretrained("/storage/ongoing/new/Open-Sora-Plan/cache_dir/mt5-xxl", cache_dir=args.cache_dir)
+    tokenizer = AutoTokenizer.from_pretrained("google/mt5-xxl", cache_dir=args.cache_dir)
     transformer = OpenSoraT2V.from_pretrained(args.model_path, cache_dir=args.cache_dir, low_cpu_mem_usage=False, 
                                               device_map=None, torch_dtype=weight_dtype)
     scheduler = EulerAncestralDiscreteScheduler()

diff --git a/scripts/text_condition/gpu/sample_t2v.sh b/scripts/text_condition/gpu/sample_t2v.sh
@@ -3,7 +3,7 @@ CUDA_VISIBLE_DEVICES=7 python opensora/sample/sample_t2v.py \
     --num_frames 29 \
     --height 480 \
     --width 640 \
-    --cache_dir "../cache_dir" \
+    --cache_dir "./cache_dir" \
     --text_encoder_name google/mt5-xxl \
     --text_prompt examples/prompt_list_0.txt \
     --ae CausalVAEModel_D4_4x8x8 \

diff --git a/scripts/text_condition/gpu/sample_t2v_ddp.sh b/scripts/text_condition/gpu/sample_t2v_ddp.sh
@@ -5,7 +5,7 @@ torchrun --nnodes=1 --nproc_per_node 8  --master_port 29503 \
     --num_frames 93 \
     --height 720 \
     --width 1280 \
-    --cache_dir "../cache_dir" \
+    --cache_dir "./cache_dir" \
     --text_encoder_name google/mt5-xxl \
     --text_prompt examples/prompt_list_0.txt \
     --ae CausalVAEModel_D4_4x8x8 \

diff --git a/scripts/text_condition/gpu/sample_t2v_sp.sh b/scripts/text_condition/gpu/sample_t2v_sp.sh
@@ -5,7 +5,7 @@ torchrun --nnodes=1 --nproc_per_node 8  --master_port 29503 \
     --num_frames 93 \
     --height 720 \
     --width 1280 \
-    --cache_dir "../cache_dir" \
+    --cache_dir "./cache_dir" \
     --text_encoder_name google/mt5-xxl \
     --text_prompt examples/prompt_list_0.txt \
     --ae CausalVAEModel_D4_4x8x8 \