sd-pixel/install.py

9 lines
340 B
Python

import launch
# This extension uses cv2 for image effects
if not launch.is_installed("opencv-python"):
launch.run_pip("install opencv-python", "opencv-python")
for dep in ['onnxruntime', 'pymatting', 'pooch', 'opencv-python']:
if not launch.is_installed(dep):
launch.run_pip(f"install {dep}", f"{dep} for pixel extension")