OzzyGT
/

pipeline_sdxl_fill

Model card Files Files and versions Community

OzzyGT HF staff commited on Sep 13

Commit

ab9a0a1

•

1 Parent(s): 0f7a0a6

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -39,7 +39,7 @@ def retrieve_timesteps(
     return timesteps, num_inference_steps
-class StableDiffusionXLPipeline(DiffusionPipeline, StableDiffusionMixin):
     model_cpu_offload_seq = "text_encoder->text_encoder_2->unet->vae"
     _optional_components = [
         "tokenizer",

     return timesteps, num_inference_steps
+class StableDiffusionXLControlNetPipeline(DiffusionPipeline, StableDiffusionMixin):
     model_cpu_offload_seq = "text_encoder->text_encoder_2->unet->vae"
     _optional_components = [
         "tokenizer",