Unverified Commit 4b868f14 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

post release 0.33.0 (#11255)



* post release

* update

* fix deprecations

* remaining

* update

---------
Co-authored-by: default avatarYiYi Xu <yixu310@gmail.com>
parent b6156aaf
...@@ -65,7 +65,7 @@ if is_wandb_available(): ...@@ -65,7 +65,7 @@ if is_wandb_available():
import wandb import wandb
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__) logger = get_logger(__name__)
......
...@@ -74,7 +74,7 @@ if is_wandb_available(): ...@@ -74,7 +74,7 @@ if is_wandb_available():
import wandb import wandb
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__) logger = get_logger(__name__)
......
...@@ -72,7 +72,7 @@ if is_wandb_available(): ...@@ -72,7 +72,7 @@ if is_wandb_available():
import wandb import wandb
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__) logger = get_logger(__name__)
......
...@@ -48,7 +48,7 @@ import diffusers ...@@ -48,7 +48,7 @@ import diffusers
from diffusers import ( from diffusers import (
AutoencoderKL, AutoencoderKL,
FlowMatchEulerDiscreteScheduler, FlowMatchEulerDiscreteScheduler,
Lumina2Text2ImgPipeline, Lumina2Pipeline,
Lumina2Transformer2DModel, Lumina2Transformer2DModel,
) )
from diffusers.optimization import get_scheduler from diffusers.optimization import get_scheduler
...@@ -72,7 +72,7 @@ if is_wandb_available(): ...@@ -72,7 +72,7 @@ if is_wandb_available():
import wandb import wandb
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__) logger = get_logger(__name__)
...@@ -898,7 +898,7 @@ def main(args): ...@@ -898,7 +898,7 @@ def main(args):
cur_class_images = len(list(class_images_dir.iterdir())) cur_class_images = len(list(class_images_dir.iterdir()))
if cur_class_images < args.num_class_images: if cur_class_images < args.num_class_images:
pipeline = Lumina2Text2ImgPipeline.from_pretrained( pipeline = Lumina2Pipeline.from_pretrained(
args.pretrained_model_name_or_path, args.pretrained_model_name_or_path,
torch_dtype=torch.bfloat16 if args.mixed_precision == "bf16" else torch.float16, torch_dtype=torch.bfloat16 if args.mixed_precision == "bf16" else torch.float16,
revision=args.revision, revision=args.revision,
...@@ -990,7 +990,7 @@ def main(args): ...@@ -990,7 +990,7 @@ def main(args):
text_encoder.to(dtype=torch.bfloat16) text_encoder.to(dtype=torch.bfloat16)
# Initialize a text encoding pipeline and keep it to CPU for now. # Initialize a text encoding pipeline and keep it to CPU for now.
text_encoding_pipeline = Lumina2Text2ImgPipeline.from_pretrained( text_encoding_pipeline = Lumina2Pipeline.from_pretrained(
args.pretrained_model_name_or_path, args.pretrained_model_name_or_path,
vae=None, vae=None,
transformer=None, transformer=None,
...@@ -1034,7 +1034,7 @@ def main(args): ...@@ -1034,7 +1034,7 @@ def main(args):
# make sure to pop weight so that corresponding model is not saved again # make sure to pop weight so that corresponding model is not saved again
weights.pop() weights.pop()
Lumina2Text2ImgPipeline.save_lora_weights( Lumina2Pipeline.save_lora_weights(
output_dir, output_dir,
transformer_lora_layers=transformer_lora_layers_to_save, transformer_lora_layers=transformer_lora_layers_to_save,
) )
...@@ -1050,7 +1050,7 @@ def main(args): ...@@ -1050,7 +1050,7 @@ def main(args):
else: else:
raise ValueError(f"unexpected save model: {model.__class__}") raise ValueError(f"unexpected save model: {model.__class__}")
lora_state_dict = Lumina2Text2ImgPipeline.lora_state_dict(input_dir) lora_state_dict = Lumina2Pipeline.lora_state_dict(input_dir)
transformer_state_dict = { transformer_state_dict = {
f"{k.replace('transformer.', '')}": v for k, v in lora_state_dict.items() if k.startswith("transformer.") f"{k.replace('transformer.', '')}": v for k, v in lora_state_dict.items() if k.startswith("transformer.")
...@@ -1473,7 +1473,7 @@ def main(args): ...@@ -1473,7 +1473,7 @@ def main(args):
if accelerator.is_main_process: if accelerator.is_main_process:
if args.validation_prompt is not None and epoch % args.validation_epochs == 0: if args.validation_prompt is not None and epoch % args.validation_epochs == 0:
# create pipeline # create pipeline
pipeline = Lumina2Text2ImgPipeline.from_pretrained( pipeline = Lumina2Pipeline.from_pretrained(
args.pretrained_model_name_or_path, args.pretrained_model_name_or_path,
transformer=accelerator.unwrap_model(transformer), transformer=accelerator.unwrap_model(transformer),
revision=args.revision, revision=args.revision,
...@@ -1503,14 +1503,14 @@ def main(args): ...@@ -1503,14 +1503,14 @@ def main(args):
transformer = transformer.to(weight_dtype) transformer = transformer.to(weight_dtype)
transformer_lora_layers = get_peft_model_state_dict(transformer) transformer_lora_layers = get_peft_model_state_dict(transformer)
Lumina2Text2ImgPipeline.save_lora_weights( Lumina2Pipeline.save_lora_weights(
save_directory=args.output_dir, save_directory=args.output_dir,
transformer_lora_layers=transformer_lora_layers, transformer_lora_layers=transformer_lora_layers,
) )
# Final inference # Final inference
# Load previous pipeline # Load previous pipeline
pipeline = Lumina2Text2ImgPipeline.from_pretrained( pipeline = Lumina2Pipeline.from_pretrained(
args.pretrained_model_name_or_path, args.pretrained_model_name_or_path,
revision=args.revision, revision=args.revision,
variant=args.variant, variant=args.variant,
......
...@@ -71,7 +71,7 @@ if is_wandb_available(): ...@@ -71,7 +71,7 @@ if is_wandb_available():
import wandb import wandb
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__) logger = get_logger(__name__)
......
...@@ -72,7 +72,7 @@ if is_wandb_available(): ...@@ -72,7 +72,7 @@ if is_wandb_available():
import wandb import wandb
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__) logger = get_logger(__name__)
......
...@@ -79,7 +79,7 @@ if is_wandb_available(): ...@@ -79,7 +79,7 @@ if is_wandb_available():
import wandb import wandb
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__) logger = get_logger(__name__)
......
...@@ -63,7 +63,7 @@ if is_wandb_available(): ...@@ -63,7 +63,7 @@ if is_wandb_available():
import wandb import wandb
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__) logger = get_logger(__name__)
......
...@@ -54,7 +54,7 @@ if is_wandb_available(): ...@@ -54,7 +54,7 @@ if is_wandb_available():
import wandb import wandb
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__) logger = get_logger(__name__)
......
...@@ -57,7 +57,7 @@ if is_wandb_available(): ...@@ -57,7 +57,7 @@ if is_wandb_available():
import wandb import wandb
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__) logger = get_logger(__name__)
......
...@@ -58,7 +58,7 @@ if is_wandb_available(): ...@@ -58,7 +58,7 @@ if is_wandb_available():
import wandb import wandb
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__, log_level="INFO") logger = get_logger(__name__, log_level="INFO")
......
...@@ -60,7 +60,7 @@ if is_wandb_available(): ...@@ -60,7 +60,7 @@ if is_wandb_available():
import wandb import wandb
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__, log_level="INFO") logger = get_logger(__name__, log_level="INFO")
......
...@@ -52,7 +52,7 @@ if is_wandb_available(): ...@@ -52,7 +52,7 @@ if is_wandb_available():
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__, log_level="INFO") logger = get_logger(__name__, log_level="INFO")
......
...@@ -46,7 +46,7 @@ from diffusers.utils import check_min_version, is_wandb_available ...@@ -46,7 +46,7 @@ from diffusers.utils import check_min_version, is_wandb_available
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__, log_level="INFO") logger = get_logger(__name__, log_level="INFO")
......
...@@ -46,7 +46,7 @@ from diffusers.utils import check_min_version, is_wandb_available ...@@ -46,7 +46,7 @@ from diffusers.utils import check_min_version, is_wandb_available
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__, log_level="INFO") logger = get_logger(__name__, log_level="INFO")
......
...@@ -51,7 +51,7 @@ if is_wandb_available(): ...@@ -51,7 +51,7 @@ if is_wandb_available():
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__, log_level="INFO") logger = get_logger(__name__, log_level="INFO")
......
...@@ -60,7 +60,7 @@ if is_wandb_available(): ...@@ -60,7 +60,7 @@ if is_wandb_available():
import wandb import wandb
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__) logger = get_logger(__name__)
......
...@@ -57,7 +57,7 @@ if is_wandb_available(): ...@@ -57,7 +57,7 @@ if is_wandb_available():
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__, log_level="INFO") logger = get_logger(__name__, log_level="INFO")
......
...@@ -49,7 +49,7 @@ from diffusers.utils import check_min_version ...@@ -49,7 +49,7 @@ from diffusers.utils import check_min_version
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
......
...@@ -56,7 +56,7 @@ if is_wandb_available(): ...@@ -56,7 +56,7 @@ if is_wandb_available():
import wandb import wandb
# Will error if the minimal version of diffusers is not installed. Remove at your own risks. # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
check_min_version("0.33.0.dev0") check_min_version("0.34.0.dev0")
logger = get_logger(__name__, log_level="INFO") logger = get_logger(__name__, log_level="INFO")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment