load captioner

mindspore-lab · Dec 24, 2024 · deb5fd1 · deb5fd1
1 parent f2e0154
commit deb5fd1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/opensora_pku/opensora/sample/sample.py b/examples/opensora_pku/opensora/sample/sample.py
@@ -12,7 +12,6 @@
 import time
 
 from opensora.npu_config import npu_config
-from opensora.sample.caption_refiner import OpenSoraCaptionRefiner
 from opensora.utils.sample_utils import get_args, prepare_pipeline, run_model_and_save_samples
 
 from mindone.utils.logger import set_logger
@@ -40,6 +39,8 @@
     pipeline = prepare_pipeline(args)  # build I2V/T2V pipeline
 
     if args.caption_refiner is not None:
+        from opensora.sample.caption_refiner import OpenSoraCaptionRefiner
+
         caption_refiner_model = OpenSoraCaptionRefiner(args.caption_refiner, dtype=ms.float16)
     else:
         caption_refiner_model = None