test

IlyasMoutawwakil · Dec 12, 2024 · 0163715 · 0163715
1 parent af8b250
commit 0163715
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py_txi/text_generation_inference.py b/py_txi/text_generation_inference.py
@@ -39,7 +39,7 @@ def __post_init__(self) -> None:
                 self.image = "ghcr.io/huggingface/text-generation-inference:3.0.1-rocm"
             else:
                 LOGGER.info("\t+ Using image version 3.0.1 for Text-Generation-Inference")
-                self.image = "ghcr.io/huggingface/text-generation-inference:3.0.1"
+                self.image = "ghcr.io/huggingface/text-generation-inference:3.0.1-intel-cpu"
 
         if is_rocm_system() and "rocm" not in self.image:
             LOGGER.warning("\t+ You are running on a ROCm AMD GPU system but using a non-ROCM image.")