Spaces:
Sleeping
Sleeping
Update h3_split_blocks.py
Browse files- 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", "
|
| 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", "
|
| 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):
|