Merge pull request #146 from ExponentialML/fix/scheduler

Fix scheduler loading

Author: ExponentialML

train.py

@@ -116,7 +116,7 @@ def create_output_folders(output_dir, config):
     return out_dir
 
 def load_primary_models(pretrained_model_path):
-    noise_scheduler = DDPMScheduler.from_pretrained(pretrained_model_path, subfolder="scheduler")
+    noise_scheduler = DDPMScheduler.from_config(pretrained_model_path, subfolder="scheduler")
     tokenizer = CLIPTokenizer.from_pretrained(pretrained_model_path, subfolder="tokenizer")
     text_encoder = CLIPTextModel.from_pretrained(pretrained_model_path, subfolder="text_encoder")
     vae = AutoencoderKL.from_pretrained(pretrained_model_path, subfolder="vae")