vrr/heretic
2
0
Fork 0
mirror of https://github.com/p-e-w/heretic.git synced 2026-07-09 17:28:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Guard against missing model card data

Browse source
This commit is contained in:
Philipp Emanuel Weidmann 2025-10-25 13:12:18 +05:30
parent e2419de016
commit 47e855d5d8
No known key found for this signature in database
GPG key ID: CE6A7756AD520E0F
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/heretic/main.py
View file

@ -20,7 +20,7 @@ from accelerate.utils import (
is_sdaa_available,
is_xpu_available,
)
from huggingface_hub import ModelCard
from huggingface_hub import ModelCard, ModelCardData
from pydantic import ValidationError
from rich.traceback import install
@ -343,6 +343,10 @@ def run():
# we can retrieve the model card.
if not Path(settings.model).exists():
card = ModelCard.load(settings.model)
if card.data is None:
card.data = ModelCardData()
if card.data.tags is None:
card.data.tags = []
card.data.tags.append("heretic")
card.data.tags.append("uncensored")
card.data.tags.append("decensored")