13 lines
326 B
Python
13 lines
326 B
Python
from .piflow_loader import PiFlowLoader
|
|
from .piflow_sampler import PiFlowSampler
|
|
from .model_sampling_piflow import ModelSamplingPiFlow
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"Load pi-Flow Model": PiFlowLoader,
|
|
"pi-Flow Sampler": PiFlowSampler,
|
|
"ModelSamplingPiFlow": ModelSamplingPiFlow,
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS']
|