stable-diffusion-aws-extension/install.py

11 lines
330 B
Python

import launch
import os
req_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), "scripts", "requirements.txt")
with open(req_file) as file:
for lib in file:
lib = lib.strip()
if not launch.is_installed(lib):
launch.run_pip(f"install {lib}", f"sd-webui-sagemaker requirement: {lib}")