From 5f249f3aefa58b1aad282d587c464d25c396d74d Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Sun, 23 Jun 2024 10:53:43 -0400
Subject: [PATCH] do not apply pag for img2img

---
 modules/pag/__init__.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/modules/pag/__init__.py b/modules/pag/__init__.py
index 484fccc13..83eba5f0b 100644
--- a/modules/pag/__init__.py
+++ b/modules/pag/__init__.py
@@ -18,6 +18,9 @@ def apply(p: processing.StableDiffusionProcessing): # pylint: disable=arguments-
         return None
     if p.pag_scale == 0:
         return
+    if sd_models.get_diffusers_task(shared.sd_model) != sd_models.DiffusersTaskType.TEXT_2_IMAGE:
+        shared.log.warning(f'PAG: pipeline={c} not implemented')
+        return None
     if detect.is_sd15(c):
         orig_pipeline = shared.sd_model
         shared.sd_model = sd_models.switch_pipe(StableDiffusionPAGPipeline, shared.sd_model)