mirror of
https://github.com/vladmandic/automatic
synced 2026-08-30 09:01:01 +02:00
Vladimir Mandic
11 lines
454 B
Python
11 lines
454 B
Python
import diffusers
|
|
from .pipeline_mage_flow import MageFlowPipeline
|
|
from .pipeline_output import MageFlowPipelineOutput
|
|
from .autoencoder_mage_vae import AutoencoderMageVAE
|
|
from .transformer_mage_flow import MageFlowTransformer2DModel
|
|
|
|
diffusers.MageFlowPipeline = MageFlowPipeline
|
|
diffusers.MageFlowPipelineOutput = MageFlowPipelineOutput
|
|
diffusers.AutoencoderMageVAE = AutoencoderMageVAE
|
|
diffusers.MageFlowTransformer2DModel = MageFlowTransformer2DModel
|