From 426713292666283f2d37e788941a8a4170f5f683 Mon Sep 17 00:00:00 2001 From: Kevin Turner <83819+keturn@users.noreply.github.com> Date: Thu, 17 Aug 2023 13:42:29 -0700 Subject: [PATCH] dep(diffusers): upgrade diffusers to 0.20 Removed `is_safetensors_available` as safetensors is now a required dependency of diffusers. --- invokeai/backend/model_management/models/vae.py | 3 +-- pyproject.toml | 4 ++-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/invokeai/backend/model_management/models/vae.py b/invokeai/backend/model_management/models/vae.py index 957a102ffb..20ce28ad86 100644 --- a/invokeai/backend/model_management/models/vae.py +++ b/invokeai/backend/model_management/models/vae.py @@ -5,7 +5,6 @@ from typing import Optional import safetensors import torch -from diffusers.utils import is_safetensors_available from omegaconf import OmegaConf from invokeai.app.services.config import InvokeAIAppConfig @@ -175,5 +174,5 @@ def _convert_vae_ckpt_and_cache( vae_config=config, image_size=image_size, ) - vae_model.save_pretrained(output_path, safe_serialization=is_safetensors_available()) + vae_model.save_pretrained(output_path, safe_serialization=True) return output_path diff --git a/pyproject.toml b/pyproject.toml index e13ae607b9..2ccbb2ac05 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -40,7 +40,7 @@ dependencies = [ "controlnet-aux>=0.0.6", "timm==0.6.13", # needed to override timm latest in controlnet_aux, see https://github.com/isl-org/ZoeDepth/issues/26 "datasets", - "diffusers[torch]~=0.19.3", + "diffusers[torch]~=0.20.0", "dnspython~=2.4.0", "dynamicprompts", "easing-functions", @@ -49,7 +49,7 @@ dependencies = [ "fastapi==0.88.0", "fastapi-events==0.8.0", "fastapi-socketio==0.0.10", - "huggingface-hub>=0.11.1", + "huggingface-hub~=0.16.4", "invisible-watermark~=0.2.0", # needed to install SDXL base and refiner using their repo_ids "matplotlib", # needed for plotting of Penner easing functions "mediapipe", # needed for "mediapipeface" controlnet model