Fix CI by upgrading and pinning dependencies (#371)

basetenlabs · Oct 30, 2024 · 13325c0 · 13325c0
1 parent b595d5f
commit 13325c0
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 8 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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