Skip to content

GitLab

Commit dd870f3f authored by Yang Yong(雍洋)'s avatar Yang Yong(雍洋) Committed by GitHub
Browse files

remove bucket_shape log (#288)

parent e3663f4b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
lightx2v/models/runners/wan/wan_audio_runner.py
View file @ dd870f3f
...@@ -122,14 +122,14 @@ def resize_image(img, resize_mode="adaptive", bucket_shape=None, fixed_area=None ...@@ -122,14 +122,14 @@ def resize_image(img, resize_mode="adaptive", bucket_shape=None, fixed_area=None
bucket_config = {} bucket_config = {}
for ratio, resolutions in bucket_shape.items(): for ratio, resolutions in bucket_shape.items():
bucket_config[float(ratio)] = np.array(resolutions, dtype=np.int64) bucket_config[float(ratio)] = np.array(resolutions, dtype=np.int64)
logger.info(f"[wan_audio] use custom bucket_shape: {bucket_config}") # logger.info(f"[wan_audio] use custom bucket_shape: {bucket_config}")
else: else:
bucket_config = { bucket_config = {
0.667: np.array([[480, 832], [544, 960], [720, 1280]], dtype=np.int64), 0.667: np.array([[480, 832], [544, 960], [720, 1280]], dtype=np.int64),
1.500: np.array([[832, 480], [960, 544], [1280, 720]], dtype=np.int64), 1.500: np.array([[832, 480], [960, 544], [1280, 720]], dtype=np.int64),
1.000: np.array([[480, 480], [576, 576], [704, 704], [960, 960]], dtype=np.int64), 1.000: np.array([[480, 480], [576, 576], [704, 704], [960, 960]], dtype=np.int64),
} }
logger.info(f"[wan_audio] use default bucket_shape: {bucket_config}") # logger.info(f"[wan_audio] use default bucket_shape: {bucket_config}")
ori_height = img.shape[-2] ori_height = img.shape[-2]
ori_weight = img.shape[-1] ori_weight = img.shape[-1]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment