[shieldgemma] fix checkpoint loading (#39348)

* fix * fix * fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2025-07-14 11:34:58 +05:00
parent a1ad9197c5
commit 66cd995618
3 changed files with 11 additions and 7 deletions
--- a/src/transformers/modeling_utils.py
+++ b/src/transformers/modeling_utils.py
@@ -240,6 +240,7 @@ VLMS = [
    "mistral3",
    "mllama",
    "paligemma",
+    "shieldgemma2",
    "qwen2vl",
    "qwen2_5_vl",
    "videollava",