File tree Expand file tree Collapse file tree 4 files changed +4
-4
lines changed Expand file tree Collapse file tree 4 files changed +4
-4
lines changed Original file line number Diff line number Diff 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 ,
Original file line number Diff line number Diff 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 ,
Original file line number Diff line number Diff 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).
Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments