From 2fb34b00b5c5daa1b60c46a5b535d30c5acf35fc Mon Sep 17 00:00:00 2001 From: jtwolfe Date: Wed, 24 Apr 2024 17:17:49 +1000 Subject: [PATCH] Incl ocv pkg for diffsusers utils (#2115) * Update diffusers.yml Signed-off-by: jtwolfe * Update diffusers-rocm.yml Signed-off-by: jtwolfe --------- Signed-off-by: jtwolfe --- backend/python/diffusers/diffusers-rocm.yml | 1 + backend/python/diffusers/diffusers.yml | 1 + 2 files changed, 2 insertions(+) diff --git a/backend/python/diffusers/diffusers-rocm.yml b/backend/python/diffusers/diffusers-rocm.yml index f261701d..97b2ce0f 100644 --- a/backend/python/diffusers/diffusers-rocm.yml +++ b/backend/python/diffusers/diffusers-rocm.yml @@ -61,4 +61,5 @@ dependencies: - urllib3==2.0.6 - zipp==3.17.0 - torch + - opencv-python prefix: /opt/conda/envs/diffusers diff --git a/backend/python/diffusers/diffusers.yml b/backend/python/diffusers/diffusers.yml index b1a7d9f9..d5d2913e 100644 --- a/backend/python/diffusers/diffusers.yml +++ b/backend/python/diffusers/diffusers.yml @@ -71,4 +71,5 @@ dependencies: - typing-extensions==4.8.0 - urllib3==2.0.6 - zipp==3.17.0 + - opencv-python prefix: /opt/conda/envs/diffusers