Skip to content

Commit 634be6e

Browse files
authored
[examples] use from_pretrained to load scheduler (huggingface#1549)
us from_pretrained to load scheduler
1 parent d1bcbf3 commit 634be6e

File tree

4 files changed

+4
-4
lines changed

4 files changed

+4
-4
lines changed

examples/dreambooth/train_dreambooth.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -498,7 +498,7 @@ def main(args):
498498
eps=args.adam_epsilon,
499499
)
500500

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

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

examples/dreambooth/train_dreambooth_inpaint.py

+1-1
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

+1-1
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

+1-1
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)