saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
src/transformers/modelcard.py
View file

@ -454,6 +454,7 @@ class TrainingSummary:
metric_mapping = infer_metric_tags_from_eval_results(self.eval_results)
metadata = {}
metadata = _insert_value(metadata, "library_name", "transformers")
metadata = _insert_values_as_list(metadata, "language", self.language)
metadata = _insert_value(metadata, "license", self.license)
if self.finetuned_from is not None and isinstance(self.finetuned_from, str) and len(self.finetuned_from) > 0:

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")