Update run.py

run.py

@@ -14,6 +14,7 @@ def main(prompt, seed, output_directory, model_path):
 def load_model(model_path):
     device = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu')
     pipe = SynGenDiffusionPipeline.from_pretrained(model_path).to(device)
+    pipe.disable_xformers_memory_efficient_attention()
 
     return pipe