rui40000andClaude Opus 4.8
1bd90c71b6
满屏文字水印（watermark_node.py）: - 文案/字体/字号/旋转角度/密度/字间距/透明度/颜色 八项可调 - 交错网格平铺后整体旋转再从中心裁切，任意角度下四角均无空白 FeyNobg 抠图（feynobg_node.py + feynobg/）: - 内嵌 nobg 推理子集（Apache-2.0），全自动去背景，输出 alpha 与去背景图 - 重写预处理去掉对 transformers>=5.4 的依赖，4.x 环境可直接使用 - 修复权重键名与 transformers 4.x 的 SwinBackbone 命名不兼容： 不处理时 958 个参数仅 405 个对得上，backbone 形同随机初始化， 模型不报错但 alpha 几乎全黑；现按环境自动重映射并严格校验， 除确定性 buffer 外任何失配都直接中止 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING, Any
|
|
|
|
from huggingface_hub import PyTorchModelHubMixin, whoami
|
|
|
|
from .utils import set_doc
|
|
|
|
if TYPE_CHECKING:
|
|
from _typeshed import DataclassInstance
|
|
|
|
|
|
class Revised_Mixin(PyTorchModelHubMixin):
|
|
@set_doc(PyTorchModelHubMixin.push_to_hub.__doc__)
|
|
def push_to_hub(
|
|
self,
|
|
repo_id: str,
|
|
*,
|
|
config: dict | DataclassInstance | None = None,
|
|
commit_message: str = "Push model using huggingface_hub.",
|
|
private: bool | None = None,
|
|
token: str | None = None,
|
|
branch: str | None = None,
|
|
create_pr: bool | None = None,
|
|
allow_patterns: list[str] | str | None = None,
|
|
ignore_patterns: list[str] | str | None = None,
|
|
delete_patterns: list[str] | str | None = None,
|
|
model_card_kwargs: dict[str, Any] | None = None,
|
|
) -> str:
|
|
if model_card_kwargs is None:
|
|
model_card_kwargs = {}
|
|
if "/" not in repo_id:
|
|
username = whoami()["name"]
|
|
repo_id = f"{username}/{repo_id}"
|
|
model_card_kwargs["repo_id"] = repo_id
|
|
return super().push_to_hub(
|
|
repo_id,
|
|
config=config,
|
|
commit_message=commit_message,
|
|
private=private,
|
|
token=token,
|
|
branch=branch,
|
|
create_pr=create_pr,
|
|
allow_patterns=allow_patterns,
|
|
ignore_patterns=ignore_patterns,
|
|
delete_patterns=delete_patterns,
|
|
model_card_kwargs=model_card_kwargs,
|
|
)
|