add first logic for from hub code download

Author: patrickvonplaten

models/vision/ddpm/modeling_ddpm.py

@@ -20,6 +20,9 @@
 
 
 class DDPM(DiffusionPipeline):
+
+    modeling_file = "modeling_ddpm.py"
+
     def __init__(self, unet, noise_scheduler):
         super().__init__()
         self.register_modules(unet=unet, noise_scheduler=noise_scheduler)

src/diffusers/pipeline_utils.py

@@ -53,8 +53,11 @@ def register_modules(self, **kwargs):
             # retrive class_name
             class_name = module.__class__.__name__
 
+            register_dict = {name: (library, class_name)}
+            register_dict["_module"] = self.__module__
+
             # save model index config
-            self.register(**{name: (library, class_name)})
+            self.register(**register_dict)
 
             # set models
             setattr(self, name, module)
@@ -84,7 +87,10 @@ def save_pretrained(self, save_directory: Union[str, os.PathLike]):
     def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.PathLike]], **kwargs):
         # use snapshot download here to get it working from from_pretrained
         cached_folder = snapshot_download(pretrained_model_name_or_path)
-        config_dict, _ = cls.get_config_dict(cached_folder)
+        config_dict, pipeline_kwargs = cls.get_config_dict(cached_folder)
+
+        module = pipeline_kwargs["_module"]
+        # TODO(Suraj) - make from hub import work
 
         init_kwargs = {}