Update app.py

app.py

@@ -8,12 +8,11 @@ import gradio as gr
 import numpy as np
 import uuid
 import spaces
-from diffusers import ConsistencyDecoderVAE, DPMSolverMultistepScheduler, Transformer2DModel, AutoencoderKL, SASolverScheduler
+from diffusers import PixArtSigmaPipeline, ConsistencyDecoderVAE, DPMSolverMultistepScheduler, Transformer2DModel, AutoencoderKL, SASolverScheduler
 import torch
 from typing import Tuple
 from datetime import datetime
 from peft import PeftModel
-from diffusers_patches import pixart_sigma_init_patched_inputs, PixArtSigmaPipeline
 
 
 DESCRIPTION = """![Logo](https://raw.githubusercontent.com/PixArt-alpha/PixArt-sigma-project/master/static/images/logo-sigma.png)
@@ -108,12 +107,6 @@ if torch.cuda.is_available():
     weight_dtype = torch.float16
     T5_token_max_length = 300
 
-    # tmp patches for diffusers PixArtSigmaPipeline Implementation
-    print(
-        "Changing _init_patched_inputs method of diffusers.models.Transformer2DModel "
-        "using scripts.diffusers_patches.pixart_sigma_init_patched_inputs")
-    setattr(Transformer2DModel, '_init_patched_inputs', pixart_sigma_init_patched_inputs)
-
     transformer = Transformer2DModel.from_pretrained(
         "PixArt-alpha/PixArt-Sigma-XL-2-1024-MS",
         subfolder='transformer',