File size: 590 Bytes
d91766b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | __all__ = [
"EditSamplerMixin",
"EditSchedulerMixin",
"TokenMergeSamplerMixin",
]
def __getattr__(name):
if name in {"EditSamplerMixin", "EditSchedulerMixin"}:
from .edit import EditSamplerMixin, EditSchedulerMixin
return {
"EditSamplerMixin": EditSamplerMixin,
"EditSchedulerMixin": EditSchedulerMixin,
}[name]
if name == "TokenMergeSamplerMixin":
from .token_merge import TokenMergeSamplerMixin
return TokenMergeSamplerMixin
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
|