fixed a bug

hellomeme/pipelines/pipline_hm5_video.py

@@ -143,8 +143,8 @@ class HM5VideoPipeline(HMPipeline):
                                 range(0, num_paded_indexes, stride)]
 
             if len(drive_idx_chunks[-1]) == 1:
-                drive_idx_chunks[-1] = drive_idx_chunks[:-1]
-                drive_idx_chunks_noise[-1] = drive_idx_chunks_noise[:-1]
+                drive_idx_chunks = drive_idx_chunks[:-1]
+                drive_idx_chunks_noise = drive_idx_chunks_noise[:-1]
 
             margin_indexes = set()
             for chunk in drive_idx_chunks:
@@ -163,7 +163,6 @@ class HM5VideoPipeline(HMPipeline):
             algorithm_type="sde-dpmsolver++",
         )
 
-
         timesteps, _ = retrieve_timesteps(scheduler, step, device, timesteps, sigmas)
 
         N, C, _, H, W = base_noise.shape

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "comfyui_hellomeme"
 description = "This repository is the official implementation of the [a/HelloMeme](https://arxiv.org/pdf/2410.22901) ComfyUI interface"
-version = "1.3.8"
+version = "1.3.9"
 license = {file = "LICENSE"}
 
 [project.urls]