mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-01-26 10:59:47 +00:00
22 lines
612 B
Python
22 lines
612 B
Python
import pkg_resources
|
|
|
|
from modules.launch_utils import run_pip
|
|
|
|
target_bitsandbytes_version = '0.45.2'
|
|
|
|
|
|
def try_install_bnb():
|
|
try:
|
|
bitsandbytes_version = pkg_resources.get_distribution('bitsandbytes').version
|
|
except Exception:
|
|
bitsandbytes_version = None
|
|
|
|
try:
|
|
if bitsandbytes_version != target_bitsandbytes_version:
|
|
run_pip(
|
|
f"install -U bitsandbytes=={target_bitsandbytes_version}",
|
|
f"bitsandbytes=={target_bitsandbytes_version}",
|
|
)
|
|
except Exception as e:
|
|
print(f'Cannot install bitsandbytes. Skipped.')
|