cuda

baseline_utils.py

@@ -82,19 +82,19 @@ def generate_comic_book(diary_text, writer_description, num_pages=4):
         cache_dir="./SDXL-Turbo"
     )
 
-    # Check for available device: CUDA, MPS, or CPU
-    if torch.cuda.is_available():
-        device = "cuda"
-        print("Using CUDA backend.")
-    elif torch.backends.mps.is_available():
-        device = "mps"
-        print("Using MPS backend.")
-    else:
-        device = "cpu"
-        print("CUDA and MPS not available. Falling back to CPU.")
+    # # Check for available device: CUDA, MPS, or CPU
+    # if torch.cuda.is_available():
+    #     device = "cuda"
+    #     print("Using CUDA backend.")
+    # elif torch.backends.mps.is_available():
+    #     device = "mps"
+    #     print("Using MPS backend.")
+    # else:
+    #     device = "cpu"
+    #     print("CUDA and MPS not available. Falling back to CPU.")
 
     # Move the model to the selected device
-    pipe = pipe.to(device)
+    pipe = pipe.to('cuda')
 
     # Create a directory to store the comic book input_images
     os.makedirs("comic_book", exist_ok=True)