From 87211c34578a97e1f9a3ef7d41d404b20a6228fc Mon Sep 17 00:00:00 2001 From: helloyongyang Date: Wed, 22 Apr 2026 07:38:57 +0000 Subject: [PATCH] update neo++ seed --- examples/neopp/neopp_dense_2k.py | 10 +++++----- lightx2v/models/runners/wan/wan_matrix_game3_runner.py | 10 ++++++++-- lightx2v/pipeline.py | 3 ++- 3 files changed, 15 insertions(+), 8 deletions(-) diff --git a/examples/neopp/neopp_dense_2k.py b/examples/neopp/neopp_dense_2k.py index 182cd93b8..ea6d46fc2 100644 --- a/examples/neopp/neopp_dense_2k.py +++ b/examples/neopp/neopp_dense_2k.py @@ -36,7 +36,7 @@ pipe.generate( seed=200, - save_result_path="/data/nvme1/yongyang/FL/LightX2V/save_results/output_lightx2v_neopp_dense_2k_0.png", + save_result_path="/data/nvme1/yongyang/kkk/LightX2V/save_results/output_lightx2v_neopp_dense_2k_0.png", target_shape=[2048, 2048], # Height, Width ) @@ -58,8 +58,8 @@ ) pipe.generate( - seed=201, - save_result_path="/data/nvme1/yongyang/FL/LightX2V/save_results/output_lightx2v_neopp_dense_2k_1.png", + seed=None, + save_result_path="/data/nvme1/yongyang/kkk/LightX2V/save_results/output_lightx2v_neopp_dense_2k_1.png", target_shape=[2048, 2048], # Height, Width ) @@ -81,7 +81,7 @@ ) pipe.generate( - seed=202, - save_result_path="/data/nvme1/yongyang/FL/LightX2V/save_results/output_lightx2v_neopp_dense_2k_2.png", + seed=None, + save_result_path="/data/nvme1/yongyang/kkk/LightX2V/save_results/output_lightx2v_neopp_dense_2k_2.png", target_shape=[2048, 2048], # Height, Width ) diff --git a/lightx2v/models/runners/wan/wan_matrix_game3_runner.py b/lightx2v/models/runners/wan/wan_matrix_game3_runner.py index 1623cc077..d053f1fb1 100644 --- a/lightx2v/models/runners/wan/wan_matrix_game3_runner.py +++ b/lightx2v/models/runners/wan/wan_matrix_game3_runner.py @@ -14,8 +14,14 @@ from diffusers.utils import deprecate from einops import rearrange from loguru import logger -from scipy.interpolate import interp1d -from scipy.spatial.transform import Rotation, Slerp + +try: + from scipy.interpolate import interp1d + from scipy.spatial.transform import Rotation, Slerp +except ImportError: + interp1d = None + Rotation = None + Slerp = None from lightx2v.models.runners.wan.wan_runner import Wan22DenseRunner, build_wan_model_with_lora from lightx2v.models.schedulers.scheduler import BaseScheduler diff --git a/lightx2v/pipeline.py b/lightx2v/pipeline.py index 4573e10d3..fbba300be 100755 --- a/lightx2v/pipeline.py +++ b/lightx2v/pipeline.py @@ -460,7 +460,8 @@ def generate( self.modify_config({"task": self.task}) input_info = init_empty_input_info(self.task, self.support_tasks) - seed_all(self.seed) + if self.seed is not None: + seed_all(self.seed) update_input_info_from_dict(input_info, self) gen_result = self.runner.run_pipeline(input_info) logger.info("Generated successfully!")