Update pipeline.py

pipeline.py

@@ -7,7 +7,7 @@ import torch
 
 class MOSDiffusionPipeline(DiffusionPipeline):
 
-    def __init__(self, reload_from_ckpt=None, base_folder=None):
+    def __init__(self, reload_from_ckpt="./qa-mdt/checkpoint_389999.ckpt", base_folder=None):
         """
         Initialize the MOS Diffusion pipeline and download the necessary files/folders.
 
@@ -26,8 +26,7 @@ class MOSDiffusionPipeline(DiffusionPipeline):
         self.reload_from_ckpt = reload_from_ckpt
         config_yaml_path = os.path.join(self.config_yaml)
         self.configs = self.load_yaml(config_yaml_path)
-        if self.reload_from_ckpt is not None:
-            self.configs["reload_from_ckpt"] = self.reload_from_ckpt
+        self.configs["reload_from_ckpt"] = self.reload_from_ckpt
         self.exp_name = os.path.basename(self.config_yaml.split(".")[0])
         self.exp_group_name = os.path.basename(os.path.dirname(self.config_yaml))