skip uninstall on experimental

This commit is contained in:
Vladimir Mandic
2023-08-12 22:16:37 +02:00
parent 69eaf4c664
commit 1cfebbb717
2 changed files with 3 additions and 3 deletions
+2 -2
View File
@@ -384,7 +384,7 @@ def check_torch():
try:
if 'xformers' in xformers_package:
install(f'--no-deps {xformers_package}', ignore=True)
else:
elif not args.experimental:
x = pkg_resources.working_set.by_key.get('xformers', None)
if x is not None:
log.warning(f'Not used, uninstalling: {x}')
@@ -435,7 +435,7 @@ def install_packages():
bitsandbytes_package = os.environ.get('BITSANDBYTES_PACKAGE', None)
if bitsandbytes_package is not None:
install(bitsandbytes_package, 'bitsandbytes', ignore=True)
else:
elif not args.experimental:
bitsandbytes_package = pkg_resources.working_set.by_key.get('bitsandbytes', None)
if bitsandbytes_package is not None:
log.warning(f'Not used, uninstalling: {bitsandbytes_package}')