diff --git a/.gitignore b/.gitignore index 8a581342d..a29980f78 100644 --- a/.gitignore +++ b/.gitignore @@ -58,6 +58,7 @@ tunableop_results*.csv !webui.sh !package.json !requirements.txt +!constraints.txt !/data !/models/VAE-approx !/models/VAE-approx/model.pt diff --git a/TODO.md b/TODO.md index c4b820105..2c035e628 100644 --- a/TODO.md +++ b/TODO.md @@ -2,7 +2,6 @@ ## Release -- Bumb packages - Implement `unload_auxiliary_models` - Release **Launcher** - Release **Enso** diff --git a/constraints.txt b/constraints.txt new file mode 100644 index 000000000..e69de29bb diff --git a/installer.py b/installer.py index 6617596d6..090bc0748 100644 --- a/installer.py +++ b/installer.py @@ -475,7 +475,7 @@ def check_diffusers(): t_start = time.time() if args.skip_all: return - target_commit = "e5aa719241f9b74d6700be3320a777799bfab70a" # diffusers commit hash + target_commit = "c02c17c6ee7ac508c56925dde4d4a3c587650dc3" # diffusers commit hash # if args.use_rocm or args.use_zluda or args.use_directml: # sha = '043ab2520f6a19fce78e6e060a68dbc947edb9f9' # lock diffusers versions for now pkg = package_spec('diffusers') @@ -1235,6 +1235,7 @@ def install_requirements(): # set environment variables controling the behavior of various libraries def set_environment(): log.debug('Setting environment tuning') + os.environ.setdefault('PIP_CONSTRAINT', os.path.abspath('constraints.txt')) os.environ.setdefault('ACCELERATE', 'True') os.environ.setdefault('ATTN_PRECISION', 'fp16') os.environ.setdefault('ClDeviceGlobalMemSizeAvailablePercent', '100') diff --git a/package.json b/package.json index e3e948c65..9fe57fd1a 100644 --- a/package.json +++ b/package.json @@ -18,7 +18,7 @@ "venv": ". venv/bin/activate", "start": ". venv/bin/activate; python launch.py --debug", "localize": "node cli/localize.js", - "packages": ". venv/bin/activate && pip install --upgrade transformers accelerate huggingface_hub safetensors tokenizers peft pytorch_lightning pylint ruff", + "packages": ". venv/bin/activate && pip install --upgrade accelerate huggingface_hub safetensors tokenizers peft pytorch_lightning pylint ruff", "format": ". venv/bin/activate && pre-commit run --all-files", "format-win": "venv\\scripts\\activate && pre-commit run --all-files", "eslint": "eslint . javascript/", diff --git a/requirements.txt b/requirements.txt index 67c45ac46..e8ca68aba 100644 --- a/requirements.txt +++ b/requirements.txt @@ -27,13 +27,13 @@ peft==0.18.1 httpx==0.28.1 requests==2.32.3 tqdm==4.67.3 -accelerate==1.12.0 +accelerate==1.13.0 einops==0.8.1 -huggingface_hub==1.5.0 +huggingface_hub==1.7.2 numpy==2.1.2 pandas==2.3.1 protobuf==6.33.5 -pytorch_lightning==2.6.0 +pytorch_lightning==2.6.1 urllib3==1.26.19 Pillow==10.4.0 timm==1.0.24