Commit 4659c635 authored by helloyongyang's avatar helloyongyang
Browse files

fix ci

parent a4818f0f
...@@ -45,11 +45,7 @@ class WanScheduler4ChangingResolution(WanScheduler): ...@@ -45,11 +45,7 @@ class WanScheduler4ChangingResolution(WanScheduler):
# 2. upsample clean noise to target shape # 2. upsample clean noise to target shape
denoised_sample_5d = denoised_sample.unsqueeze(0) # (C,T,H,W) -> (1,C,T,H,W) denoised_sample_5d = denoised_sample.unsqueeze(0) # (C,T,H,W) -> (1,C,T,H,W)
clean_noise = torch.nn.functional.interpolate( clean_noise = torch.nn.functional.interpolate(denoised_sample_5d, size=(self.config.target_shape[1], self.config.target_shape[2], self.config.target_shape[3]), mode="trilinear")
denoised_sample_5d,
size=(self.config.target_shape[1], self.config.target_shape[2], self.config.target_shape[3]),
mode='trilinear'
)
clean_noise = clean_noise.squeeze(0) # (1,C,T,H,W) -> (C,T,H,W) clean_noise = clean_noise.squeeze(0) # (1,C,T,H,W) -> (C,T,H,W)
# 3. add noise to clean noise # 3. add noise to clean noise
......
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