From 13325c0450c40a30cb4a48d07ddc96c61de4401d Mon Sep 17 00:00:00 2001
From: joostinyi <63941848+joostinyi@users.noreply.github.com>
Date: Tue, 29 Oct 2024 17:28:58 -0700
Subject: [PATCH] Fix CI by upgrading and pinning dependencies (#371)

---
 playground-v2-aesthetic/config.yaml                  | 1 +
 stable-diffusion/stable-diffusion-xl-1.0/config.yaml | 8 ++++----
 whisper/faster-whisper-v2/config.yaml                | 5 +++--
 whisper/faster-whisper-v3/config.yaml                | 5 +++--
 4 files changed, 11 insertions(+), 8 deletions(-)

diff --git a/playground-v2-aesthetic/config.yaml b/playground-v2-aesthetic/config.yaml
index e180ac40..3bc6743d 100644
--- a/playground-v2-aesthetic/config.yaml
+++ b/playground-v2-aesthetic/config.yaml
@@ -23,6 +23,7 @@ requirements:
 - transformers==4.35.2
 - diffusers==0.24.0
 - accelerate==0.25.0
+- huggingface-hub==0.19.4
 resources:
   accelerator: A10G
   cpu: '2'
diff --git a/stable-diffusion/stable-diffusion-xl-1.0/config.yaml b/stable-diffusion/stable-diffusion-xl-1.0/config.yaml
index d03b9df1..85df00f3 100644
--- a/stable-diffusion/stable-diffusion-xl-1.0/config.yaml
+++ b/stable-diffusion/stable-diffusion-xl-1.0/config.yaml
@@ -28,12 +28,12 @@ model_metadata:
 model_name: Stable Diffusion XL
 python_version: py39
 requirements:
-- transformers
-- accelerate
-- safetensors
+- transformers==4.34.0
+- accelerate==0.23.0
+- safetensors==0.4.0
 - git+https://github.com/basetenlabs/diffusers.git@9a353290b1497023d4745a719ec02c50f680499a
 - invisible-watermark>=0.2.0
-- xformers
+- xformers==0.0.22
 resources:
   accelerator: A10G
   cpu: '2'
diff --git a/whisper/faster-whisper-v2/config.yaml b/whisper/faster-whisper-v2/config.yaml
index 7bade046..da604f48 100644
--- a/whisper/faster-whisper-v2/config.yaml
+++ b/whisper/faster-whisper-v2/config.yaml
@@ -15,8 +15,9 @@ model_metadata:
 model_name: Faster Whisper v2
 python_version: py39
 requirements:
-- torch==2.1.0
-- faster-whisper==0.10.0
+- torch==2.1.1
+- faster-whisper==1.0.3
+- ctranslate2==4.4.0
 resources:
   accelerator: T4
   cpu: 500m
diff --git a/whisper/faster-whisper-v3/config.yaml b/whisper/faster-whisper-v3/config.yaml
index d631845c..3571edd3 100644
--- a/whisper/faster-whisper-v3/config.yaml
+++ b/whisper/faster-whisper-v3/config.yaml
@@ -15,8 +15,9 @@ model_metadata:
 model_name: Faster Whisper v3
 python_version: py39
 requirements:
-- torch==2.1.0
-- faster-whisper==0.10.0
+- torch==2.1.1
+- faster-whisper==1.0.3
+- ctranslate2==4.4.0
 resources:
   accelerator: T4
   cpu: 500m