Skip to content

Commit c5a0fb1

Browse files
authored
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
1 parent 20a45dd commit c5a0fb1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

onnxruntime/python/tools/transformers/models/stable_diffusion/requirements.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,10 +9,11 @@ packaging
99
protobuf==3.20.3
1010
psutil
1111
sympy
12-
controlnet_aux==0.0.7
12+
controlnet_aux==0.0.9
1313
# The following are for SDXL
1414
optimum==1.20.0
1515
safetensors
1616
invisible_watermark
1717
# newer version of opencv-python migth encounter module 'cv2.dnn' has no attribute 'DictValue' error
1818
opencv-python==4.8.0.74
19+
opencv-python-headless==4.8.0.74

0 commit comments

Comments
 (0)