mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-12-29 10:10:35 +08:00
67 lines
1.7 KiB
Python
67 lines
1.7 KiB
Python
# 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": # train command will fallback to sft command
|
|
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
|