Simple wrapper to match the output format expected by pipeline
| 3 | |
| 4 | |
| 5 | class DecoderOutput: |
| 6 | """Simple wrapper to match the output format expected by pipeline""" |
| 7 | def __init__(self, sample): |
| 8 | self.sample = sample |
| 9 | |
| 10 | |
| 11 | def setup_tiny_vae(pipeline, model_type="wan2.1", tiny_vae_path=None, parallel_decode=False, need_scaled=False): |