update

canonicalize/pipeline_canonicalize.py

@@ -394,7 +394,10 @@ class CanonicalizationPipeline(DiffusionPipeline):
             batch_size = image.shape[0]
         # Define call parameters
         batch_size = 1 if isinstance(prompt, str) else len(prompt)
-        device = self._execution_device
+        # device = self._execution_device
+        device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+        import sys
+        print(f"PIPELINE Using device!!!!!!!!!!!!: {device}", file=sys.stderr)
         # here `guidance_scale` is defined analog to the guidance weight `w` of equation (2)
         # of the Imagen paper: https://arxiv.org/pdf/2205.11487.pdf . `guidance_scale = 1`
         # corresponds to doing no classifier free guidance.