| # Copyright 2025 the LlamaFactory team. | |
| # | |
| # Licensed under the Apache License, Version 2.0 (the "License"); | |
| # you may not use this file except in compliance with the License. | |
| # You may obtain a copy of the License at | |
| # | |
| # http://www.apache.org/licenses/LICENSE-2.0 | |
| # | |
| # Unless required by applicable law or agreed to in writing, software | |
| # distributed under the License is distributed on an "AS IS" BASIS, | |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
| # See the License for the specific language governing permissions and | |
| # limitations under the License. | |
| import sys | |
| from ..extras.env import VERSION, print_env | |
| USAGE = ( | |
| "-" * 70 | |
| + "\n" | |
| + "| Usage: |\n" | |
| + "| llamafactory-cli sft -h: train models |\n" | |
| + "| llamafactory-cli version: show version info |\n" | |
| + "| Hint: You can use `lmf` as a shortcut for `llamafactory-cli`. |\n" | |
| + "-" * 70 | |
| ) | |
| WELCOME = ( | |
| "-" * 58 | |
| + "\n" | |
| + f"| Welcome to LLaMA Factory, version {VERSION}" | |
| + " " * (21 - len(VERSION)) | |
| + "|\n|" | |
| + " " * 56 | |
| + "|\n" | |
| + "| Project page: https://github.com/hiyouga/LLaMA-Factory |\n" | |
| + "-" * 58 | |
| ) | |
| def launch(): | |
| command = sys.argv.pop(1) if len(sys.argv) > 1 else "help" | |
| if command == "sft": | |
| from .trainers.sft_trainer import run_sft | |
| run_sft() | |
| elif command == "env": | |
| print_env() | |
| elif command == "version": | |
| print(WELCOME) | |
| elif command == "help": | |
| print(USAGE) | |
| else: | |
| print(f"Unknown command: {command}.\n{USAGE}") | |
| if __name__ == "__main__": | |
| pass | |