Apply suggestions from code review

Co-authored-by: Will Berman <[email protected]>

Author: sayakpaul

examples/instruct_pix2pix/train_instruct_pix2pix.py

@@ -907,7 +907,7 @@ def collate_fn(examples):
                 original_image = download_image(args.val_image_url)
                 edited_images = []
                 with torch.autocast(
-                    str(accelerator.device).replace(":0", ""), enabled=accelerator.mixed_precision == "fp16"
+                    "cuda", enabled=accelerator.mixed_precision == "fp16"
                 ):
                     for _ in range(args.num_validation_images):
                         edited_images.append(
@@ -963,7 +963,7 @@ def collate_fn(examples):
         if args.validation_prompt is not None:
             edited_images = []
             pipeline = pipeline.to(accelerator.device)
-            with torch.autocast(str(accelerator.device).replace(":0", "")):
+            with torch.autocast("cuda"):
                 for _ in range(args.num_validation_images):
                     edited_images.append(
                         pipeline(