fix tests

Author: sayakpaul

src/diffusers/pipelines/pipeline_utils.py

@@ -313,7 +313,7 @@ def is_saveable_module(name, value):
                 dduf_file_path = os.path.join(save_directory, dduf_file)
                 dir_to_archive = os.path.join(save_directory, pipeline_component_name)
                 if os.path.isdir(dir_to_archive):
-                    export_folder_as_dduf(dduf_file_path, dir_to_archive, append=True, retain_base_folder=True)
+                    export_folder_as_dduf(dduf_file_path, dir_to_archive)
                     shutil.rmtree(dir_to_archive)
 
         # finally save the config

tests/pipelines/test_pipelines_common.py

@@ -1905,6 +1905,8 @@ def test_StableDiffusionMixin_component(self):
     # @pytest.mark.xfail(condition=not os.getenv("RUN_DDUF_TEST", False), strict=True)
     # Should consider guarding the test with proper transformers and huggingface_hub versions.
     def test_save_load_dduf(self):
+        from huggingface_hub import export_folder_as_dduf
+
         components = self.get_dummy_components()
         pipe = self.pipeline_class(**components)
         pipe = pipe.to(torch_device)
@@ -1917,8 +1919,10 @@ def test_save_load_dduf(self):
         pipeline_out = pipe(**inputs).images
 
         with tempfile.TemporaryDirectory() as tmpdir:
-            dduf_filename = f"{pipe.__class__.__name__.lower()}.dduf"
-            pipe.save_pretrained(tmpdir, dduf_file=dduf_filename)
+            dduf_filename = os.path.join(tmpdir, f"{pipe.__class__.__name__.lower()}.dduf")
+            pipe.save_pretrained(tmpdir, safe_serialization=True)
+            export_folder_as_dduf(dduf_filename, folder_path=tmpdir)
+            print(f"{os.listdir(tmpdir)=}")
             loaded_pipe = self.pipeline_class.from_pretrained(tmpdir, dduf_file=dduf_filename).to(torch_device)
 
         inputs["generator"] = torch.manual_seed(0)