Use import alias

pull/2426/head
Adam J. Stewart 2025-01-22 10:08:18 +01:00 committed by Ross Wightman
parent 4de1abf837
commit d81da93c16
1 changed files with 17 additions and 29 deletions

View File

@ -1,30 +1,18 @@
from .version import __version__
from .layers import is_scriptable, is_exportable, set_scriptable, set_exportable
from .models import (
create_model,
list_models,
list_pretrained,
is_model,
list_modules,
model_entrypoint,
is_model_pretrained,
get_pretrained_cfg,
get_pretrained_cfg_value,
from .version import __version__ as __version__
from .layers import (
is_scriptable as is_scriptable,
is_exportable as is_exportable,
set_scriptable as set_scriptable,
set_exportable as set_exportable,
)
from .models import (
create_model as create_model,
list_models as list_models,
list_pretrained as list_pretrained,
is_model as is_model,
list_modules as list_modules,
model_entrypoint as model_entrypoint,
is_model_pretrained as is_model_pretrained,
get_pretrained_cfg as get_pretrained_cfg,
get_pretrained_cfg_value as get_pretrained_cfg_value,
)
__all__ = [
"__version__",
"is_scriptable",
"is_exportable",
"set_scriptable",
"set_exportable",
"create_model",
"list_models",
"list_pretrained",
"is_model",
"list_modules",
"model_entrypoint",
"is_model_pretrained",
"get_pretrained_cfg",
"get_pretrained_cfg_value",
]