dagloop5 commited on
Commit
fe57fe7
·
verified ·
1 Parent(s): 2a1e01b

Update h3_split_blocks.py

Browse files
Files changed (1) hide show
  1. h3_split_blocks.py +2 -5
h3_split_blocks.py CHANGED
@@ -15,7 +15,6 @@ frame count is aligned to `17 * n + 5` before the call, since that arithmetic li
15
  import torch
16
 
17
  from diffusers.modular_pipelines.minimax_h3.before_encoder import MiniMaxH3ResizeStep, MiniMaxH3Ref2VASetupStep
18
- from diffusers.modular_pipelines.minimax_h3.decoders import MiniMaxH3AfterDenoiseStep
19
  from diffusers.modular_pipelines.minimax_h3.encoders import (
20
  MiniMaxH3KeyframeVaeEncoderStep,
21
  MiniMaxH3Ref2VAReferenceEncoderStep,
@@ -159,10 +158,9 @@ class MiniMaxH3GeneratorBlocks(SequentialPipelineBlocks):
159
  MiniMaxH3SplitBeforeEncodeStep,
160
  MiniMaxH3SplitVaeEncoderStep,
161
  MiniMaxH3SplitDenoiseStep,
162
- MiniMaxH3AfterDenoiseStep,
163
  MiniMaxH3DecodeStep,
164
  ]
165
- block_names = ["resize", "vae_encoder", "denoise", "after_denoise", "decode"]
166
 
167
  @property
168
  def description(self):
@@ -222,10 +220,9 @@ class MiniMaxH3Ref2VAGeneratorBlocks(SequentialPipelineBlocks):
222
  MiniMaxH3Ref2VASetupStep,
223
  MiniMaxH3Ref2VAReferenceEncoderStep,
224
  MiniMaxH3Ref2VACoreDenoiseStep,
225
- MiniMaxH3AfterDenoiseStep,
226
  MiniMaxH3DecodeStep,
227
  ]
228
- block_names = ["setup", "reference_encoder", "denoise", "after_denoise", "decode"]
229
 
230
  @property
231
  def description(self):
 
15
  import torch
16
 
17
  from diffusers.modular_pipelines.minimax_h3.before_encoder import MiniMaxH3ResizeStep, MiniMaxH3Ref2VASetupStep
 
18
  from diffusers.modular_pipelines.minimax_h3.encoders import (
19
  MiniMaxH3KeyframeVaeEncoderStep,
20
  MiniMaxH3Ref2VAReferenceEncoderStep,
 
158
  MiniMaxH3SplitBeforeEncodeStep,
159
  MiniMaxH3SplitVaeEncoderStep,
160
  MiniMaxH3SplitDenoiseStep,
 
161
  MiniMaxH3DecodeStep,
162
  ]
163
+ block_names = ["resize", "vae_encoder", "denoise", "decode"]
164
 
165
  @property
166
  def description(self):
 
220
  MiniMaxH3Ref2VASetupStep,
221
  MiniMaxH3Ref2VAReferenceEncoderStep,
222
  MiniMaxH3Ref2VACoreDenoiseStep,
 
223
  MiniMaxH3DecodeStep,
224
  ]
225
+ block_names = ["setup", "reference_encoder", "denoise", "decode"]
226
 
227
  @property
228
  def description(self):