tipsv2/load_one.py

"""Load a single model and print all missing and unexpected weights.

Swap AutoModel for AutoBackbone / Tipsv2VisionModel / Tipsv2TextModel and
change REPO as needed.
"""

from transformers import AutoModel, AutoBackbone, Tipsv2VisionModel, Tipsv2TextModel, AutoModelForDepthEstimation, AutoModelForNormalEstimation, AutoModelForSemanticSegmentation, Tipsv2DptModel, Tipsv2DptForDepthEstimation, Tipsv2DptForNormalEstimation, Tipsv2DptForSemanticSegmentation

REPO = "google/tipsv2-b14-dpt"
REVISION = "refs/pr/1"

model, loading_info = AutoModel.from_pretrained(
    REPO,
    trust_remote_code=True,
    output_loading_info=True,
)

print(f"Loaded {REPO} -> {type(model).__name__}")