[pipeline utils] sanitize pretrained_model_name_or_path (#5173)

Make sure the repo_id is valid before sending it to huggingface_hub to get a more understandable error message. Re #5110 Co-authored-by: Patrick von Platen <[email protected]>
huggingface · Sep 25, 2023 · dfdf85d · dfdf85d
1 parent 539846a
commit dfdf85d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/diffusers/pipelines/pipeline_utils.py b/src/diffusers/pipelines/pipeline_utils.py
@@ -1033,6 +1033,10 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
         # 1. Download the checkpoints and configs
         # use snapshot download here to get it working from from_pretrained
         if not os.path.isdir(pretrained_model_name_or_path):
+            if pretrained_model_name_or_path.count("/") > 1: 
+                raise ValueError(                                                                                                                                                                                                          
+                   f"The provided pretrained_model_name_or_path \"{pretrained_model_name_or_path}\""
+                   " is neither a valid local path nor a valid repo id. Please check the parameter.")
             cached_folder = cls.download(
                 pretrained_model_name_or_path,
                 cache_dir=cache_dir,