File size: 1,643 Bytes
2fdce3c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
import argparse
import logging
import setup_common
errors = 0 # Define the 'errors' variable before using it
log = logging.getLogger('sd')
# ANSI escape code for yellow color
YELLOW = '\033[93m'
RESET_COLOR = '\033[0m'
def main_menu(platform_requirements_file, show_stdout: bool = False, no_run_accelerate: bool = False):
log.info("Installing python dependencies. This could take a few minutes as it downloads files.")
log.info("If this operation ever runs too long, you can rerun this script in verbose mode to check.")
setup_common.check_repo_version()
setup_common.check_python()
# Upgrade pip if needed
setup_common.install('--upgrade pip')
setup_common.install_requirements(platform_requirements_file, check_no_verify_flag=False, show_stdout=show_stdout)
if not no_run_accelerate:
setup_common.configure_accelerate(run_accelerate=False)
if __name__ == '__main__':
setup_common.ensure_base_requirements()
setup_common.setup_logging()
parser = argparse.ArgumentParser()
parser.add_argument('--platform-requirements-file', dest='platform_requirements_file', default='requirements_linux.txt', help='Path to the platform-specific requirements file')
parser.add_argument('--show_stdout', dest='show_stdout', action='store_true', help='Whether to show stdout during installation')
parser.add_argument('--no_run_accelerate', dest='no_run_accelerate', action='store_true', help='Whether to not run accelerate config')
args = parser.parse_args()
main_menu(args.platform_requirements_file, show_stdout=args.show_stdout, no_run_accelerate=args.no_run_accelerate)
|