Skip to content

Commit

Permalink
squash! convert : add --print-supported-models option
Browse files Browse the repository at this point in the history
Fix flake8 error.
  • Loading branch information
danbev committed Jan 10, 2025
1 parent beae794 commit fd2672b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions convert_hf_to_gguf.py
Original file line number Diff line number Diff line change
Expand Up @@ -481,7 +481,7 @@ def func(modelcls: AnyModel) -> AnyModel:
@classmethod
def print_registered_models(cls):
for name in cls._model_classes.keys():
print(f"- {name}")
logger.error(f"- {name}")

@classmethod
def from_model_architecture(cls, arch: str) -> type[Model]:
Expand Down Expand Up @@ -5005,7 +5005,7 @@ def main() -> None:
args = parse_args()

if args.print_supported_models:
print("Supported models:")
logger.error("Supported models:")
Model.print_registered_models()
sys.exit(0)

Expand Down

0 comments on commit fd2672b

Please sign in to comment.