@@ -35,10 +35,13 @@ from diffusers import PixArtAlphaPipeline
3535from tgate import TgatePixArtLoader
3636
3737pipe = PixArtAlphaPipeline.from_pretrained(" PixArt-alpha/PixArt-XL-2-1024-MS" , torch_dtype = torch.float16)
38+
39+ gate_step = 8
40+ inference_step = 25
3841pipe = TgatePixArtLoader(
3942 pipe,
40- gate_step = 8 ,
41- num_inference_steps = 25 ,
43+ gate_step = gate_step ,
44+ num_inference_steps = inference_step ,
4245).to(" cuda" )
4346
4447image = pipe.tgate(
@@ -56,6 +59,7 @@ Accelerate `StableDiffusionXLPipeline` with T-GATE:
5659import torch
5760from diffusers import StableDiffusionXLPipeline
5861from diffusers import DPMSolverMultistepScheduler
62+ from tgate import TgateSDXLLoader
5963
6064pipe = StableDiffusionXLPipeline.from_pretrained(
6165 " stabilityai/stable-diffusion-xl-base-1.0" ,
@@ -65,7 +69,6 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
6569)
6670pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
6771
68- from tgate import TgateSDXLLoader
6972gate_step = 10
7073inference_step = 25
7174pipe = TgateSDXLLoader(
@@ -89,6 +92,7 @@ Accelerate `StableDiffusionXLPipeline` with [DeepCache](https://github.com/horse
8992import torch
9093from diffusers import StableDiffusionXLPipeline
9194from diffusers import DPMSolverMultistepScheduler
95+ from tgate import TgateSDXLDeepCacheLoader
9296
9397pipe = StableDiffusionXLPipeline.from_pretrained(
9498 " stabilityai/stable-diffusion-xl-base-1.0" ,
@@ -98,7 +102,6 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
98102)
99103pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
100104
101- from tgate import TgateSDXLDeepCacheLoader
102105gate_step = 10
103106inference_step = 25
104107pipe = TgateSDXLDeepCacheLoader(
@@ -123,6 +126,7 @@ import torch
123126from diffusers import StableDiffusionXLPipeline
124127from diffusers import UNet2DConditionModel, LCMScheduler
125128from diffusers import DPMSolverMultistepScheduler
129+ from tgate import TgateSDXLLoader
126130
127131unet = UNet2DConditionModel.from_pretrained(
128132 " latent-consistency/lcm-sdxl" ,
@@ -137,7 +141,6 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
137141)
138142pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
139143
140- from tgate import TgateSDXLLoader
141144gate_step = 1
142145inference_step = 4
143146pipe = TgateSDXLLoader(
0 commit comments