diff --git a/installer.py b/installer.py index 9c77a1cf2..a07444abf 100644 --- a/installer.py +++ b/installer.py @@ -761,7 +761,8 @@ def install_packages(): clip_package = os.environ.get('CLIP_PACKAGE', "git+https://github.com/openai/CLIP.git") install(clip_package, 'clip', quiet=True) tensorflow_package = os.environ.get('TENSORFLOW_PACKAGE', 'tensorflow==2.13.0') - install(tensorflow_package, 'tensorflow-rocm' if 'rocm' in tensorflow_package else 'tensorflow', ignore=True, quiet=True) + if tensorflow_package != '': + install(tensorflow_package, 'tensorflow-rocm' if 'rocm' in tensorflow_package else 'tensorflow', ignore=True, quiet=True) bitsandbytes_package = os.environ.get('BITSANDBYTES_PACKAGE', None) if bitsandbytes_package is not None: install(bitsandbytes_package, 'bitsandbytes', ignore=True, quiet=True) diff --git a/modules/loader.py b/modules/loader.py index 04e6dfca2..1fe3ba81e 100644 --- a/modules/loader.py +++ b/modules/loader.py @@ -10,14 +10,8 @@ from modules import timer, errors initialized = False errors.install() logging.getLogger("DeepSpeed").disabled = True -# os.environ.setdefault('OMP_NUM_THREADS', 1) -# os.environ.setdefault('MKL_NUM_THREADS', 1) - -# import tensorflow as tf # pylint: disable=C0411 import torch # pylint: disable=C0411 - -# torch.set_num_threads(1) try: import intel_extension_for_pytorch as ipex # pylint: disable=import-error, unused-import errors.log.debug(f'Load IPEX=={ipex.__version__}')