Skip to content

Commit e89f4a0

Browse files
patil-surajThomas Capelle
authored andcommitted
[examples] use from_pretrained to load scheduler (huggingface#1549)
us from_pretrained to load scheduler
1 parent 45aa346 commit e89f4a0

File tree

4 files changed

+4
-4
lines changed

4 files changed

+4
-4
lines changed

examples/dreambooth/train_dreambooth.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -499,7 +499,7 @@ def main(args):
499499
eps=args.adam_epsilon,
500500
)
501501

502-
noise_scheduler = DDPMScheduler.from_config(args.pretrained_model_name_or_path, subfolder="scheduler")
502+
noise_scheduler = DDPMScheduler.from_pretrained(args.pretrained_model_name_or_path, subfolder="scheduler")
503503

504504
train_dataset = DreamBoothDataset(
505505
instance_data_root=args.instance_data_dir,

examples/dreambooth/train_dreambooth_inpaint.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -500,7 +500,7 @@ def main():
500500
eps=args.adam_epsilon,
501501
)
502502

503-
noise_scheduler = DDPMScheduler.from_config(args.pretrained_model_name_or_path, subfolder="scheduler")
503+
noise_scheduler = DDPMScheduler.from_pretrained(args.pretrained_model_name_or_path, subfolder="scheduler")
504504

505505
train_dataset = DreamBoothDataset(
506506
instance_data_root=args.instance_data_dir,

examples/text_to_image/train_text_to_image.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -392,7 +392,7 @@ def main():
392392
weight_decay=args.adam_weight_decay,
393393
eps=args.adam_epsilon,
394394
)
395-
noise_scheduler = DDPMScheduler.from_config(args.pretrained_model_name_or_path, subfolder="scheduler")
395+
noise_scheduler = DDPMScheduler.from_pretrained(args.pretrained_model_name_or_path, subfolder="scheduler")
396396

397397
# Get the datasets: you can either provide your own training and evaluation files (see below)
398398
# or specify a Dataset from the hub (the dataset will be downloaded automatically from the datasets Hub).

examples/textual_inversion/textual_inversion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -466,7 +466,7 @@ def main():
466466
eps=args.adam_epsilon,
467467
)
468468

469-
noise_scheduler = DDPMScheduler.from_config(args.pretrained_model_name_or_path, subfolder="scheduler")
469+
noise_scheduler = DDPMScheduler.from_pretrained(args.pretrained_model_name_or_path, subfolder="scheduler")
470470

471471
train_dataset = TextualInversionDataset(
472472
data_root=args.train_data_dir,

0 commit comments

Comments
 (0)