SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Automatically add transformers tag to the modelcard (#32623)

Browse Source 
* Automatically add `transformers` tag to the modelcard

* Specify library_name and test
This commit is contained in:
Lysandre Debut
2024-08-13 07:59:01 +02:00
committed by GitHub
parent a29eabd0eb
commit 29c3a0fa01
2 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/utils/test_model_card.py
View File

@@ -19,7 +19,7 @@ import os
import tempfile
import unittest
from transformers.modelcard import ModelCard
from transformers.modelcard import ModelCard, TrainingSummary
class ModelCardTester(unittest.TestCase):
@@ -82,3 +82,8 @@ class ModelCardTester(unittest.TestCase):
model_card_second = ModelCard.from_pretrained(tmpdirname)
self.assertEqual(model_card_second.to_dict(), model_card_first.to_dict())
def test_model_summary_modelcard_base_metadata(self):
metadata = TrainingSummary("Model name").create_metadata()
self.assertTrue("library_name" in metadata)
self.assertTrue(metadata["library_name"] == "transformers")