update model repo to fix tests

Author: philschmid

.github/workflows/docker-build-action.yaml

@@ -63,6 +63,7 @@ jobs:
           push: true
           context: ${{ inputs.context }}
           build-args: ${{ inputs.build_args }}
+          target: base
           file:  ${{ inputs.context }}/${{ inputs.dockerfile }}
           tags: ${{ inputs.repository }}/${{ inputs.image }}:sha-${{ env.GITHUB_SHA_SHORT }},${{ inputs.repository }}/${{ inputs.image }}:latest
 

setup.py

@@ -31,8 +31,8 @@
 extras = {}
 
 extras["st"] = ["sentence_transformers==2.7.0"]
-extras["diffusers"] = ["diffusers==0.26.3", "accelerate==0.30.1"]
-extras["torch"] = ["torch==2.2.2", "torchvision", "torchaudio"]
+extras["diffusers"] = ["diffusers==0.26.3", "accelerate==0.27.2"]
+extras["torch"] = ["torch==2.2.0", "torchvision", "torchaudio"]
 extras["test"] = [
     "pytest==7.2.1",
     "pytest-xdist",

tests/unit/test_diffusers.py

@@ -15,7 +15,7 @@
 def test_get_diffusers_pipeline():
     with tempfile.TemporaryDirectory() as tmpdirname:
         storage_dir = _load_repository_from_hf(
-            "hf-internal-testing/tiny-stable-diffusion-torch",
+            "echarlaix/tiny-random-stable-diffusion-xl",
             tmpdirname,
             framework="pytorch"
         )
@@ -28,7 +28,7 @@ def test_get_diffusers_pipeline():
 def test_pipe_on_gpu():
     with tempfile.TemporaryDirectory() as tmpdirname:
         storage_dir = _load_repository_from_hf(
-            "hf-internal-testing/tiny-stable-diffusion-torch",
+            "echarlaix/tiny-random-stable-diffusion-xl",
             tmpdirname,
             framework="pytorch"
         )
@@ -44,7 +44,7 @@ def test_pipe_on_gpu():
 def test_text_to_image_task():
     with tempfile.TemporaryDirectory() as tmpdirname:
         storage_dir = _load_repository_from_hf(
-            "hf-internal-testing/tiny-stable-diffusion-torch",
+            "echarlaix/tiny-random-stable-diffusion-xl",
             tmpdirname,
             framework="pytorch"
         )