Update ramalama/compose.py

approach excludes images for other GPU backends like ROCm or custom builds. Please update the check to support a wider range of GPU-enabled images.

Signed-off-by: Abhishek Bongale <abhishekbongale@outlook.com>

Co-authored-by: sourcery-ai[bot] <58596630+sourcery-ai[bot]@users.noreply.github.com>

Author: abhibongale

ramalama/compose.py

@@ -142,7 +142,8 @@ def _gen_environment(self) -> str:
         return env_spec
 
     def _gen_gpu_deployment(self) -> str:
-        if "cuda" not in self.image:
+        gpu_keywords = ["cuda", "rocm", "gpu"]
+        if not any(keyword in self.image.lower() for keyword in gpu_keywords):
             return ""
 
         return """\