Attention slicing is deprecated and cannot really be the memory bottleneck anymore

pipelines/img2img.py

@@ -112,10 +112,6 @@ class Pipeline:
         if device.type != "mps":
             self.pipe.unet.to(memory_format=torch.channels_last)
 
-        # check if computer has less than 64GB of RAM using sys or os
-        if psutil.virtual_memory().total < 64 * 1024**3:
-            self.pipe.enable_attention_slicing()
-
         if args.torch_compile:
             print("Running torch compile")
             self.pipe.unet = torch.compile(