Skip to content

GitLab

"test/vscode:/vscode.git/clone" did not exist on "0dcb3496cf3e274386272e0a4430282f9ddf1169"
Commit d50b8884 authored by helloyongyang's avatar helloyongyang
Browse files

fix bug

parent 08be2ab9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lightx2v/models/schedulers/wan/audio/scheduler.py
View file @ d50b8884
...@@ -8,7 +8,7 @@ from lightx2v.models.schedulers.scheduler import BaseScheduler ...@@ -8,7 +8,7 @@ from lightx2v.models.schedulers.scheduler import BaseScheduler
from lightx2v.utils.envs import * from lightx2v.utils.envs import *
def unsqueeze_to_ndim(in_tensor: Tensor, tgt_n_dim: int): def unsqueeze_to_ndim(in_tensor, tgt_n_dim):
if in_tensor.ndim > tgt_n_dim: if in_tensor.ndim > tgt_n_dim:
warnings.warn(f"the given tensor of shape {in_tensor.shape} is expected to unsqueeze to {tgt_n_dim}, the original tensor will be returned") warnings.warn(f"the given tensor of shape {in_tensor.shape} is expected to unsqueeze to {tgt_n_dim}, the original tensor will be returned")
return in_tensor return in_tensor
......
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