github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fixed deprecation of lumina pipeline error

Browse Source
This commit is contained in:
Jaret Burkett
2025-07-29 08:26:51 -06:00
parent ca7c5c950b
commit f453e28ea3
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
toolkit/stable_diffusion_model.py
View File

@@ -49,7 +49,7 @@ from diffusers import StableDiffusionPipeline, StableDiffusionXLPipeline, T2IAda
StableDiffusionXLImg2ImgPipeline, LCMScheduler, Transformer2DModel, AutoencoderTiny, ControlNetModel, \ StableDiffusionXLImg2ImgPipeline, LCMScheduler, Transformer2DModel, AutoencoderTiny, ControlNetModel, \
StableDiffusionXLControlNetPipeline, StableDiffusionControlNetPipeline, StableDiffusion3Pipeline, \ StableDiffusionXLControlNetPipeline, StableDiffusionControlNetPipeline, StableDiffusion3Pipeline, \
StableDiffusion3Img2ImgPipeline, PixArtSigmaPipeline, AuraFlowPipeline, AuraFlowTransformer2DModel, FluxPipeline, \ StableDiffusion3Img2ImgPipeline, PixArtSigmaPipeline, AuraFlowPipeline, AuraFlowTransformer2DModel, FluxPipeline, \
FluxTransformer2DModel, FlowMatchEulerDiscreteScheduler, SD3Transformer2DModel, Lumina2Text2ImgPipeline, \ FluxTransformer2DModel, FlowMatchEulerDiscreteScheduler, SD3Transformer2DModel, Lumina2Pipeline, \
FluxControlPipeline, Lumina2Transformer2DModel FluxControlPipeline, Lumina2Transformer2DModel
import diffusers import diffusers
from diffusers import \ from diffusers import \
@@ -886,7 +886,7 @@ class StableDiffusion:
flush() flush()
self.print_and_status_update("Making pipe") self.print_and_status_update("Making pipe")
pipe: Lumina2Text2ImgPipeline = Lumina2Text2ImgPipeline( pipe: Lumina2Pipeline = Lumina2Pipeline(
scheduler=scheduler, scheduler=scheduler,
text_encoder=None, text_encoder=None,
tokenizer=tokenizer, tokenizer=tokenizer,
@@ -1256,7 +1256,7 @@ class StableDiffusion:
pipeline.watermark = None pipeline.watermark = None
elif self.is_lumina2: elif self.is_lumina2:
pipeline = Lumina2Text2ImgPipeline( pipeline = Lumina2Pipeline(
vae=self.vae, vae=self.vae,
transformer=self.unet, transformer=self.unet,
text_encoder=self.text_encoder, text_encoder=self.text_encoder,
@@ -1580,7 +1580,7 @@ class StableDiffusion:
**extra **extra
).images[0] ).images[0]
elif self.is_lumina2: elif self.is_lumina2:
pipeline: Lumina2Text2ImgPipeline = pipeline pipeline: Lumina2Pipeline = pipeline
img = pipeline( img = pipeline(
prompt_embeds=conditional_embeds.text_embeds, prompt_embeds=conditional_embeds.text_embeds,