File size: 399 Bytes
3d79eb3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
#
# We import all the model types in order to populate the model type registry
from lcm.models.base_lcm.loader import BASE_LCM_MODEL_TYPE
from lcm.models.two_tower_diffusion_lcm.loader import (
TWO_TOWER_DIFFUSION_LCM_MODEL_TYPE,
)
__all__ = [
"BASE_LCM_MODEL_TYPE",
"TWO_TOWER_DIFFUSION_LCM_MODEL_TYPE",
]
|