Spaces:
Build error
Build error
File size: 2,489 Bytes
11c2c17 |
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 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
import argparse
import logging
import setup_common
import os
import shutil
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 configure_accelerate():
script_dir = os.path.dirname(os.path.abspath(__file__))
cache_dir = "/root/.cache/huggingface/accelerate"
log.info("Configuring accelerate...")
os.makedirs(cache_dir, exist_ok=True)
config_file_src = os.path.join(script_dir, "config_files", "accelerate", "runpod.yaml")
config_file_dest = os.path.join(cache_dir, "default_config.yaml")
shutil.copyfile(config_file_src, config_file_dest)
def setup_environment():
# Get the directory the script is run from
script_dir = os.path.dirname(os.path.abspath(__file__))
# Install tk and python3.10-venv
log.info("Install tk and python3.10-venv...")
subprocess.run(['apt', 'update', '-y'])
subprocess.run(['apt', 'install', '-y', 'python3-tk', 'python3.10-venv'])
# Check if the venv folder doesn't exist
venv_dir = os.path.join(script_dir, 'venv')
if not os.path.exists(venv_dir):
log.info("Creating venv...")
subprocess.run(['python3', '-m', 'venv', venv_dir])
# Activate the virtual environment
log.info("Activate venv...")
activate_script = os.path.join(venv_dir, 'bin', 'activate')
activate_command = f'source "{activate_script}" || exit 1'
subprocess.run(activate_command, shell=True, executable='/bin/bash')
def main_menu(platform_requirements_file):
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=True)
configure_accelerate()
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_runpod.txt', help='Path to the platform-specific requirements file')
args = parser.parse_args()
main_menu(args.platform_requirements_file)
|