Skip to content

Commit 4b868f1

Browse files
sayakpaulyiyixuxu
sayakpaul
and
yiyixuxu
authored
post release 0.33.0 (#11255)
* post release * update * fix deprecations * remaining * update --------- Co-authored-by: YiYi Xu <[email protected]>
1 parent b6156aa commit 4b868f1

File tree

52 files changed

+60
-75
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

52 files changed

+60
-75
lines changed

‎examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@
7474
import wandb
7575

7676
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
77-
check_min_version("0.33.0.dev0")
77+
check_min_version("0.34.0.dev0")
7878

7979
logger = get_logger(__name__)
8080

‎examples/advanced_diffusion_training/train_dreambooth_lora_sd15_advanced.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@
7373

7474

7575
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
76-
check_min_version("0.33.0.dev0")
76+
check_min_version("0.34.0.dev0")
7777

7878
logger = get_logger(__name__)
7979

‎examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@
8080
import wandb
8181

8282
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
83-
check_min_version("0.33.0.dev0")
83+
check_min_version("0.34.0.dev0")
8484

8585
logger = get_logger(__name__)
8686

‎examples/cogvideo/train_cogvideox_image_to_video_lora.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@
6161
import wandb
6262

6363
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
64-
check_min_version("0.33.0.dev0")
64+
check_min_version("0.34.0.dev0")
6565

6666
logger = get_logger(__name__)
6767

‎examples/cogvideo/train_cogvideox_lora.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@
5252
import wandb
5353

5454
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
55-
check_min_version("0.33.0.dev0")
55+
check_min_version("0.34.0.dev0")
5656

5757
logger = get_logger(__name__)
5858

‎examples/cogview4-control/train_control_cogview4.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@
5959
import wandb
6060

6161
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
62-
check_min_version("0.33.0.dev0")
62+
check_min_version("0.34.0.dev0")
6363

6464
logger = get_logger(__name__)
6565

‎examples/community/marigold_depth_estimation.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@
4343

4444

4545
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
46-
check_min_version("0.33.0.dev0")
46+
check_min_version("0.34.0.dev0")
4747

4848

4949
class MarigoldDepthOutput(BaseOutput):

‎examples/consistency_distillation/train_lcm_distill_lora_sd_wds.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@
7373
import wandb
7474

7575
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
76-
check_min_version("0.33.0.dev0")
76+
check_min_version("0.34.0.dev0")
7777

7878
logger = get_logger(__name__)
7979

‎examples/consistency_distillation/train_lcm_distill_lora_sdxl.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@
6666
import wandb
6767

6868
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
69-
check_min_version("0.33.0.dev0")
69+
check_min_version("0.34.0.dev0")
7070

7171
logger = get_logger(__name__)
7272

‎examples/consistency_distillation/train_lcm_distill_lora_sdxl_wds.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@
7979
import wandb
8080

8181
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
82-
check_min_version("0.33.0.dev0")
82+
check_min_version("0.34.0.dev0")
8383

8484
logger = get_logger(__name__)
8585

‎examples/consistency_distillation/train_lcm_distill_sd_wds.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@
7272
import wandb
7373

7474
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
75-
check_min_version("0.33.0.dev0")
75+
check_min_version("0.34.0.dev0")
7676

7777
logger = get_logger(__name__)
7878

‎examples/consistency_distillation/train_lcm_distill_sdxl_wds.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@
7878
import wandb
7979

8080
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
81-
check_min_version("0.33.0.dev0")
81+
check_min_version("0.34.0.dev0")
8282

8383
logger = get_logger(__name__)
8484

‎examples/controlnet/train_controlnet.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@
6060
import wandb
6161

6262
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
63-
check_min_version("0.33.0.dev0")
63+
check_min_version("0.34.0.dev0")
6464

6565
logger = get_logger(__name__)
6666

‎examples/controlnet/train_controlnet_flax.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@
6060
import wandb
6161

6262
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
63-
check_min_version("0.33.0.dev0")
63+
check_min_version("0.34.0.dev0")
6464

6565
logger = logging.getLogger(__name__)
6666

‎examples/controlnet/train_controlnet_flux.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@
5151
FlowMatchEulerDiscreteScheduler,
5252
FluxTransformer2DModel,
5353
)
54-
from diffusers.models.controlnet_flux import FluxControlNetModel
54+
from diffusers.models.controlnets.controlnet_flux import FluxControlNetModel
5555
from diffusers.optimization import get_scheduler
5656
from diffusers.pipelines.flux.pipeline_flux_controlnet import FluxControlNetPipeline
5757
from diffusers.training_utils import compute_density_for_timestep_sampling, free_memory
@@ -65,7 +65,7 @@
6565
import wandb
6666

6767
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
68-
check_min_version("0.33.0.dev0")
68+
check_min_version("0.34.0.dev0")
6969

7070
logger = get_logger(__name__)
7171
if is_torch_npu_available():

‎examples/controlnet/train_controlnet_sd3.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@
6161
import wandb
6262

6363
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
64-
check_min_version("0.33.0.dev0")
64+
check_min_version("0.34.0.dev0")
6565

6666
logger = get_logger(__name__)
6767

‎examples/controlnet/train_controlnet_sdxl.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@
6161
import wandb
6262

6363
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
64-
check_min_version("0.33.0.dev0")
64+
check_min_version("0.34.0.dev0")
6565

6666
logger = get_logger(__name__)
6767
if is_torch_npu_available():

‎examples/custom_diffusion/train_custom_diffusion.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@
6363

6464

6565
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
66-
check_min_version("0.33.0.dev0")
66+
check_min_version("0.34.0.dev0")
6767

6868
logger = get_logger(__name__)
6969

‎examples/dreambooth/train_dreambooth.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@
6363
import wandb
6464

6565
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
66-
check_min_version("0.33.0.dev0")
66+
check_min_version("0.34.0.dev0")
6767

6868
logger = get_logger(__name__)
6969

‎examples/dreambooth/train_dreambooth_flax.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
3535

3636

3737
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
38-
check_min_version("0.33.0.dev0")
38+
check_min_version("0.34.0.dev0")
3939

4040
# Cache compiled models across invocations of this script.
4141
cc.initialize_cache(os.path.expanduser("~/.cache/jax/compilation_cache"))

‎examples/dreambooth/train_dreambooth_flux.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@
6565
import wandb
6666

6767
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
68-
check_min_version("0.33.0.dev0")
68+
check_min_version("0.34.0.dev0")
6969

7070
logger = get_logger(__name__)
7171

‎examples/dreambooth/train_dreambooth_lora.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@
7474
import wandb
7575

7676
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
77-
check_min_version("0.33.0.dev0")
77+
check_min_version("0.34.0.dev0")
7878

7979
logger = get_logger(__name__)
8080

‎examples/dreambooth/train_dreambooth_lora_flux.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@
7272
import wandb
7373

7474
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
75-
check_min_version("0.33.0.dev0")
75+
check_min_version("0.34.0.dev0")
7676

7777
logger = get_logger(__name__)
7878

‎examples/dreambooth/train_dreambooth_lora_lumina2.py

+9-9
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@
4848
from diffusers import (
4949
AutoencoderKL,
5050
FlowMatchEulerDiscreteScheduler,
51-
Lumina2Text2ImgPipeline,
51+
Lumina2Pipeline,
5252
Lumina2Transformer2DModel,
5353
)
5454
from diffusers.optimization import get_scheduler
@@ -72,7 +72,7 @@
7272
import wandb
7373

7474
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
75-
check_min_version("0.33.0.dev0")
75+
check_min_version("0.34.0.dev0")
7676

7777
logger = get_logger(__name__)
7878

@@ -898,7 +898,7 @@ def main(args):
898898
cur_class_images = len(list(class_images_dir.iterdir()))
899899

900900
if cur_class_images < args.num_class_images:
901-
pipeline = Lumina2Text2ImgPipeline.from_pretrained(
901+
pipeline = Lumina2Pipeline.from_pretrained(
902902
args.pretrained_model_name_or_path,
903903
torch_dtype=torch.bfloat16 if args.mixed_precision == "bf16" else torch.float16,
904904
revision=args.revision,
@@ -990,7 +990,7 @@ def main(args):
990990
text_encoder.to(dtype=torch.bfloat16)
991991

992992
# Initialize a text encoding pipeline and keep it to CPU for now.
993-
text_encoding_pipeline = Lumina2Text2ImgPipeline.from_pretrained(
993+
text_encoding_pipeline = Lumina2Pipeline.from_pretrained(
994994
args.pretrained_model_name_or_path,
995995
vae=None,
996996
transformer=None,
@@ -1034,7 +1034,7 @@ def save_model_hook(models, weights, output_dir):
10341034
# make sure to pop weight so that corresponding model is not saved again
10351035
weights.pop()
10361036

1037-
Lumina2Text2ImgPipeline.save_lora_weights(
1037+
Lumina2Pipeline.save_lora_weights(
10381038
output_dir,
10391039
transformer_lora_layers=transformer_lora_layers_to_save,
10401040
)
@@ -1050,7 +1050,7 @@ def load_model_hook(models, input_dir):
10501050
else:
10511051
raise ValueError(f"unexpected save model: {model.__class__}")
10521052

1053-
lora_state_dict = Lumina2Text2ImgPipeline.lora_state_dict(input_dir)
1053+
lora_state_dict = Lumina2Pipeline.lora_state_dict(input_dir)
10541054

10551055
transformer_state_dict = {
10561056
f"{k.replace('transformer.', '')}": v for k, v in lora_state_dict.items() if k.startswith("transformer.")
@@ -1473,7 +1473,7 @@ def get_sigmas(timesteps, n_dim=4, dtype=torch.float32):
14731473
if accelerator.is_main_process:
14741474
if args.validation_prompt is not None and epoch % args.validation_epochs == 0:
14751475
# create pipeline
1476-
pipeline = Lumina2Text2ImgPipeline.from_pretrained(
1476+
pipeline = Lumina2Pipeline.from_pretrained(
14771477
args.pretrained_model_name_or_path,
14781478
transformer=accelerator.unwrap_model(transformer),
14791479
revision=args.revision,
@@ -1503,14 +1503,14 @@ def get_sigmas(timesteps, n_dim=4, dtype=torch.float32):
15031503
transformer = transformer.to(weight_dtype)
15041504
transformer_lora_layers = get_peft_model_state_dict(transformer)
15051505

1506-
Lumina2Text2ImgPipeline.save_lora_weights(
1506+
Lumina2Pipeline.save_lora_weights(
15071507
save_directory=args.output_dir,
15081508
transformer_lora_layers=transformer_lora_layers,
15091509
)
15101510

15111511
# Final inference
15121512
# Load previous pipeline
1513-
pipeline = Lumina2Text2ImgPipeline.from_pretrained(
1513+
pipeline = Lumina2Pipeline.from_pretrained(
15141514
args.pretrained_model_name_or_path,
15151515
revision=args.revision,
15161516
variant=args.variant,

‎examples/dreambooth/train_dreambooth_lora_sana.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@
7171
import wandb
7272

7373
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
74-
check_min_version("0.33.0.dev0")
74+
check_min_version("0.34.0.dev0")
7575

7676
logger = get_logger(__name__)
7777

‎examples/dreambooth/train_dreambooth_lora_sd3.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@
7272
import wandb
7373

7474
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
75-
check_min_version("0.33.0.dev0")
75+
check_min_version("0.34.0.dev0")
7676

7777
logger = get_logger(__name__)
7878

‎examples/dreambooth/train_dreambooth_lora_sdxl.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@
7979
import wandb
8080

8181
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
82-
check_min_version("0.33.0.dev0")
82+
check_min_version("0.34.0.dev0")
8383

8484
logger = get_logger(__name__)
8585

‎examples/dreambooth/train_dreambooth_sd3.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@
6363
import wandb
6464

6565
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
66-
check_min_version("0.33.0.dev0")
66+
check_min_version("0.34.0.dev0")
6767

6868
logger = get_logger(__name__)
6969

‎examples/flux-control/train_control_flux.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@
5454
import wandb
5555

5656
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
57-
check_min_version("0.33.0.dev0")
57+
check_min_version("0.34.0.dev0")
5858

5959
logger = get_logger(__name__)
6060

‎examples/flux-control/train_control_lora_flux.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@
5757
import wandb
5858

5959
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
60-
check_min_version("0.33.0.dev0")
60+
check_min_version("0.34.0.dev0")
6161

6262
logger = get_logger(__name__)
6363

‎examples/instruct_pix2pix/train_instruct_pix2pix.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@
5858
import wandb
5959

6060
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
61-
check_min_version("0.33.0.dev0")
61+
check_min_version("0.34.0.dev0")
6262

6363
logger = get_logger(__name__, log_level="INFO")
6464

‎examples/instruct_pix2pix/train_instruct_pix2pix_sdxl.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@
6060
import wandb
6161

6262
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
63-
check_min_version("0.33.0.dev0")
63+
check_min_version("0.34.0.dev0")
6464

6565
logger = get_logger(__name__, log_level="INFO")
6666

‎examples/kandinsky2_2/text_to_image/train_text_to_image_decoder.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@
5252

5353

5454
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
55-
check_min_version("0.33.0.dev0")
55+
check_min_version("0.34.0.dev0")
5656

5757
logger = get_logger(__name__, log_level="INFO")
5858

‎examples/kandinsky2_2/text_to_image/train_text_to_image_lora_decoder.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@
4646

4747

4848
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
49-
check_min_version("0.33.0.dev0")
49+
check_min_version("0.34.0.dev0")
5050

5151
logger = get_logger(__name__, log_level="INFO")
5252

‎examples/kandinsky2_2/text_to_image/train_text_to_image_lora_prior.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@
4646

4747

4848
# Will error if the minimal version of diffusers is not installed. Remove at your own risks.
49-
check_min_version("0.33.0.dev0")
49+
check_min_version("0.34.0.dev0")
5050

5151
logger = get_logger(__name__, log_level="INFO")
5252

0 commit comments

Comments
 (0)