From 9132ce7c58e8431aceb3ef0e56fa8f8e9af39607 Mon Sep 17 00:00:00 2001
From: Wentian <94900022+WentianZhang-ML@users.noreply.github.com>
Date: Thu, 18 Apr 2024 00:37:24 +0800
Subject: [PATCH] [Docs] Update TGATE in section `optimization`. (#7698)

Update tgate.md
---
 docs/source/en/optimization/tgate.md | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/docs/source/en/optimization/tgate.md b/docs/source/en/optimization/tgate.md
index 0b536a215fc0..d208ddfa8411 100644
--- a/docs/source/en/optimization/tgate.md
+++ b/docs/source/en/optimization/tgate.md
@@ -35,10 +35,13 @@ from diffusers import PixArtAlphaPipeline
 from tgate import TgatePixArtLoader
 
 pipe = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-XL-2-1024-MS", torch_dtype=torch.float16)
+
+gate_step = 8
+inference_step = 25
 pipe = TgatePixArtLoader(
        pipe,
-       gate_step=8,
-       num_inference_steps=25,
+       gate_step=gate_step,
+       num_inference_steps=inference_step,
 ).to("cuda")
 
 image = pipe.tgate(
@@ -56,6 +59,7 @@ Accelerate `StableDiffusionXLPipeline` with T-GATE:
 import torch
 from diffusers import StableDiffusionXLPipeline
 from diffusers import DPMSolverMultistepScheduler
+from tgate import TgateSDXLLoader
 
 pipe = StableDiffusionXLPipeline.from_pretrained(
             "stabilityai/stable-diffusion-xl-base-1.0",
@@ -65,7 +69,6 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
 )
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 
-from tgate import TgateSDXLLoader
 gate_step = 10
 inference_step = 25
 pipe = TgateSDXLLoader(
@@ -89,6 +92,7 @@ Accelerate `StableDiffusionXLPipeline` with [DeepCache](https://github.com/horse
 import torch
 from diffusers import StableDiffusionXLPipeline
 from diffusers import DPMSolverMultistepScheduler
+from tgate import TgateSDXLDeepCacheLoader
 
 pipe = StableDiffusionXLPipeline.from_pretrained(
             "stabilityai/stable-diffusion-xl-base-1.0",
@@ -98,7 +102,6 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
 )
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 
-from tgate import TgateSDXLDeepCacheLoader
 gate_step = 10
 inference_step = 25
 pipe = TgateSDXLDeepCacheLoader(
@@ -123,6 +126,7 @@ import torch
 from diffusers import StableDiffusionXLPipeline
 from diffusers import UNet2DConditionModel, LCMScheduler
 from diffusers import DPMSolverMultistepScheduler
+from tgate import TgateSDXLLoader
 
 unet = UNet2DConditionModel.from_pretrained(
     "latent-consistency/lcm-sdxl",
@@ -137,7 +141,6 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
 )
 pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
 
-from tgate import TgateSDXLLoader
 gate_step = 1
 inference_step = 4
 pipe = TgateSDXLLoader(