squash! convert : add --print-supported-models option
Fix flake8 error.
This commit is contained in:
parent
beae79455b
commit
fd2672b952
1 changed files with 2 additions and 2 deletions
|
@ -481,7 +481,7 @@ class Model:
|
||||||
@classmethod
|
@classmethod
|
||||||
def print_registered_models(cls):
|
def print_registered_models(cls):
|
||||||
for name in cls._model_classes.keys():
|
for name in cls._model_classes.keys():
|
||||||
print(f"- {name}")
|
logger.error(f"- {name}")
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_model_architecture(cls, arch: str) -> type[Model]:
|
def from_model_architecture(cls, arch: str) -> type[Model]:
|
||||||
|
@ -5005,7 +5005,7 @@ def main() -> None:
|
||||||
args = parse_args()
|
args = parse_args()
|
||||||
|
|
||||||
if args.print_supported_models:
|
if args.print_supported_models:
|
||||||
print("Supported models:")
|
logger.error("Supported models:")
|
||||||
Model.print_registered_models()
|
Model.print_registered_models()
|
||||||
sys.exit(0)
|
sys.exit(0)
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue