Remove unused imports

quic · Oct 18, 2024 · b21ebab · b21ebab
1 parent aff9740
commit b21ebab
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 47 deletions.
diff --git a/plugins/stable-diffusion-webui/qairt_accelerate/qairt_sd_controlnet_pipeline.py b/plugins/stable-diffusion-webui/qairt_accelerate/qairt_sd_controlnet_pipeline.py
@@ -6,38 +6,22 @@
 #
 # =============================================================================
 
-import time
 from PIL import Image
 import os
-import shutil
 import cv2
 import numpy as np
 import torch
 from transformers import CLIPTokenizer
-from diffusers import DPMSolverMultistepScheduler
-from modules import paths, shared, modelloader
+from modules import paths
 from qai_appbuilder import (
     QNNContext,
-    QNNContextProc,
-    QNNShareMemory,
     Runtime,
     LogLevel,
     ProfilingLevel,
     PerfProfile,
     QNNConfig,
-    timer,
 )
-from diffusers import UNet2DConditionModel
 from diffusers.models.embeddings import get_timestep_embedding, TimestepEmbedding
-from diffusers import (
-    DDIMScheduler,
-    DPMSolverMultistepScheduler,
-    EulerAncestralDiscreteScheduler,
-    EulerDiscreteScheduler,
-    HeunDiscreteScheduler,
-    LMSDiscreteScheduler,
-    PNDMScheduler,
-)
 import qairt_constants as consts
 from pipeline_utils import StableDiffusionInput, download_qualcomm_models_hf, UpscalerPipeline, QPipeline, set_scheduler
 from modules.safe import unsafe_torch_load as load

diff --git a/plugins/stable-diffusion-webui/qairt_accelerate/qairt_sd_pipeline.py b/plugins/stable-diffusion-webui/qairt_accelerate/qairt_sd_pipeline.py
@@ -6,16 +6,12 @@
 #
 # =============================================================================
 
-import time
 from PIL import Image
 import os
-import shutil
-import cv2
 import numpy as np
 import torch
 from transformers import CLIPTokenizer
-from diffusers import DPMSolverMultistepScheduler
-from modules import paths, shared, modelloader
+from modules import paths
 from qai_appbuilder import (
     QNNContext,
     Runtime,
@@ -25,17 +21,7 @@
     QNNConfig,
     timer,
 )
-from diffusers import UNet2DConditionModel
 from diffusers.models.embeddings import get_timestep_embedding, TimestepEmbedding
-from diffusers import (
-    DDIMScheduler,
-    DPMSolverMultistepScheduler,
-    EulerAncestralDiscreteScheduler,
-    EulerDiscreteScheduler,
-    HeunDiscreteScheduler,
-    LMSDiscreteScheduler,
-    PNDMScheduler,
-)
 import qairt_constants as consts
 from pipeline_utils import StableDiffusionInput, UpscalerPipeline, QPipeline, set_scheduler
 

diff --git a/plugins/stable-diffusion-webui/qairt_accelerate/qairt_sd_unclip_pipeline.py b/plugins/stable-diffusion-webui/qairt_accelerate/qairt_sd_unclip_pipeline.py
@@ -6,37 +6,29 @@
 #
 # =============================================================================
 
-import time
 from PIL import Image
-import PIL
 import os
-import sys
-import shutil
-import cv2
 import numpy as np
 import torch
 from skimage.transform import resize
 from modules.safe import unsafe_torch_load as load
-from transformers import CLIPTokenizer, CLIPImageProcessor, CLIPVisionModelWithProjection, CLIPTextModel
+from transformers import CLIPTokenizer, CLIPImageProcessor
 # from modules import paths, shared, modelloader
 from qai_appbuilder import (
     QNNContext,
-    QNNContextProc,
-    QNNShareMemory,
     Runtime,
     LogLevel,
     ProfilingLevel,
     PerfProfile,
     QNNConfig,
     timer,
 )
-from diffusers import UNet2DConditionModel, DDPMScheduler
-from diffusers.pipelines.stable_diffusion.stable_unclip_image_normalizer import StableUnCLIPImageNormalizer
-from diffusers.models.embeddings import get_timestep_embedding, TimestepEmbedding, Timesteps
+from diffusers import DDPMScheduler
+from diffusers.models.embeddings import get_timestep_embedding, TimestepEmbedding
 import install
 import common_utils as utils
 import qairt_constants as consts
-from pipeline_utils import StableDiffusionInput, download_qualcomm_models_hf, UpscalerPipeline, QPipeline, set_scheduler
+from pipeline_utils import StableDiffusionInput, UpscalerPipeline, QPipeline, set_scheduler
 
 
 class TextEncoder(QNNContext):

diff --git a/plugins/stable-diffusion-webui/qairt_accelerate/scripts/qairt_accelerate.py b/plugins/stable-diffusion-webui/qairt_accelerate/scripts/qairt_accelerate.py
@@ -97,9 +97,6 @@ def run(self, p: StableDiffusionProcessing, qnn_model_name, enable_upscale, upsc
         time_start = time.time()
 
         user_seed = get_fixed_seed(p.seed)
-        supported_samplers = ["DPM++ 2M"]
-        # if p.sampler_name not in supported_samplers:
-        #     p.sampler_name = "DPM++ 2M"
 
         sd_input = StableDiffusionInput(
             self.is_txt2img,