Updare CUDA

Image3DProcessor.py

@@ -26,7 +26,9 @@ from gaussian_renderer import render_predicted
 
 class Image3DProcessor:
     def __init__(self, model_cfg_path, model_repo_id, model_filename):
+        self.use_cuda = torch.cuda.is_available()
         self.device = torch.device("cuda" if self.use_cuda else "cpu")
+        print("Image3DProcessor Device: ", self.device)
         # Load model configuration
         self.model_cfg = OmegaConf.load(model_cfg_path)
         

SDXLImageGenerator.py

@@ -9,7 +9,7 @@ class SDXLImageGenerator:
         # Set proper device based on cuda availability
         self.device = torch.device("cuda" if self.use_cuda else "cpu")
 
-        print("CUDA: ", self.device)
+        print("SDXLImageGenerator Device: ", self.device)
 
         # Load the pipeline
         self.pipe = DiffusionPipeline.from_pretrained(