Spaces:
Running
on
Zero
Running
on
Zero
Update pipeline_minimax_remover.py
Browse files
pipeline_minimax_remover.py
CHANGED
|
@@ -195,4 +195,4 @@ class Minimax_Remover_Pipeline(DiffusionPipeline):
|
|
| 195 |
video = self.vae.decode(latents, return_dict=False)[0]
|
| 196 |
video = self.video_processor.postprocess_video(video, output_type=output_type)
|
| 197 |
|
| 198 |
-
return WanPipelineOutput(frames=video)
|
|
|
|
| 195 |
video = self.vae.decode(latents, return_dict=False)[0]
|
| 196 |
video = self.video_processor.postprocess_video(video, output_type=output_type)
|
| 197 |
|
| 198 |
+
return WanPipelineOutput(frames=video)
|