From ac73c05ed69402c5f004758f02b0caaed97b7876 Mon Sep 17 00:00:00 2001
From: Linh Vu <50764990+lvu5@users.noreply.github.com>
Date: Sat, 16 Aug 2025 00:01:08 +0400
Subject: [PATCH] Update sd_turbo.py

Add code to check if cuda is available
---
 engines/sd_turbo.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/engines/sd_turbo.py b/engines/sd_turbo.py
index 043cc6e..810934a 100644
--- a/engines/sd_turbo.py
+++ b/engines/sd_turbo.py
@@ -15,7 +15,9 @@ def get_sd_turbo_model():
 
     pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/sd-turbo", torch_dtype=torch.float16,
                                                      variant="fp16")
-    pipe.to("cuda")
+    # checking if cuda is available before moving to cuda
+    if torch.cuda.is_available():
+        pipe.to("cuda")
     return pipe