Remove modelcards dependency (#2050)

* Switch to huggingface_hub.ModelCard

* Remove modelcards dependency in favor of Jinja2

src/diffusers/utils/import_utils.py

@@ -21,6 +21,7 @@ import sys
 from collections import OrderedDict
 from typing import Union
 
+from huggingface_hub.utils import is_jinja_available  # noqa: F401
 from packaging import version
 from packaging.version import Version, parse
 
@@ -146,14 +147,6 @@ except importlib_metadata.PackageNotFoundError:
     _unidecode_available = False
 
 
-_modelcards_available = importlib.util.find_spec("modelcards") is not None
-try:
-    _modelcards_version = importlib_metadata.version("modelcards")
-    logger.debug(f"Successfully imported modelcards version {_modelcards_version}")
-except importlib_metadata.PackageNotFoundError:
-    _modelcards_available = False
-
-
 _onnxruntime_version = "N/A"
 _onnx_available = importlib.util.find_spec("onnxruntime") is not None
 if _onnx_available:
@@ -253,10 +246,6 @@ def is_unidecode_available():
     return _unidecode_available
 
 
-def is_modelcards_available():
-    return _modelcards_available
-
-
 def is_onnx_available():
     return _onnx_available
 

tests/test_hub_utils.py

+# coding=utf-8
+# Copyright 2023 HuggingFace Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+import unittest
+from pathlib import Path
+from tempfile import TemporaryDirectory
+from unittest.mock import Mock, patch
+
+import diffusers.utils.hub_utils
+
+
+class CreateModelCardTest(unittest.TestCase):
+    @patch("diffusers.utils.hub_utils.get_full_repo_name")
+    def test_create_model_card(self, repo_name_mock: Mock) -> None:
+        repo_name_mock.return_value = "full_repo_name"
+        with TemporaryDirectory() as tmpdir:
+            # Dummy args values
+            args = Mock()
+            args.output_dir = tmpdir
+            args.local_rank = 0
+            args.hub_token = "hub_token"
+            args.dataset_name = "dataset_name"
+            args.learning_rate = 0.01
+            args.train_batch_size = 100000
+            args.eval_batch_size = 10000
+            args.gradient_accumulation_steps = 0.01
+            args.adam_beta1 = 0.02
+            args.adam_beta2 = 0.03
+            args.adam_weight_decay = 0.0005
+            args.adam_epsilon = 0.000001
+            args.lr_scheduler = 1
+            args.lr_warmup_steps = 10
+            args.ema_inv_gamma = 0.001
+            args.ema_power = 0.1
+            args.ema_max_decay = 0.2
+            args.mixed_precision = True
+
+            # Model card mush be rendered and saved
+            diffusers.utils.hub_utils.create_model_card(args, model_name="model_name")
+            self.assertTrue((Path(tmpdir) / "README.md").is_file())