From 690a9fcacf95f23b7fc13006a48f80823d14870e Mon Sep 17 00:00:00 2001 From: Yi Zhang Date: Tue, 15 Oct 2024 21:13:52 +0800 Subject: [PATCH] Fix big models exception caused by timm upgrade (#22442) ### Description Today, stable diffusion stage failed due to there's a upgrade in timm. controlnet_aux depends on it. And its latest version limit the timm version less than 0.6.7. So upgrading controlnet_aux can solve it. And controlnet_aux uses opencv-python-headless, pin opencv-python-headless to 4.8.0.74 too. ### Motivation and Context --- .../transformers/models/stable_diffusion/requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/onnxruntime/python/tools/transformers/models/stable_diffusion/requirements.txt b/onnxruntime/python/tools/transformers/models/stable_diffusion/requirements.txt index de242e77cdb2e..5080737516c53 100644 --- a/onnxruntime/python/tools/transformers/models/stable_diffusion/requirements.txt +++ b/onnxruntime/python/tools/transformers/models/stable_diffusion/requirements.txt @@ -9,10 +9,11 @@ packaging protobuf==3.20.3 psutil sympy -controlnet_aux==0.0.7 +controlnet_aux==0.0.9 # The following are for SDXL optimum==1.20.0 safetensors invisible_watermark # newer version of opencv-python migth encounter module 'cv2.dnn' has no attribute 'DictValue' error opencv-python==4.8.0.74 +opencv-python-headless==4.8.0.74