Commit 563f2ff
authored
fix failure of tests/models/shieldgemma2/test_modeling_shieldgemma2.p… (#42022)
* fix failure of tests/models/shieldgemma2/test_modeling_shieldgemma2.py::ShieldGemma2IntegrationTest::test_model
Signed-off-by: Wang, Yi <yi.a.wang@intel.com>
* add common kwargs in processing_shieldgemma2.py
---------
Signed-off-by: Wang, Yi <yi.a.wang@intel.com>1 parent 6f479d5 commit 563f2ff
File tree
1 file changed
+4
-0
lines changed- src/transformers/models/shieldgemma2
1 file changed
+4
-0
lines changedLines changed: 4 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
126 | 126 | | |
127 | 127 | | |
128 | 128 | | |
| 129 | + | |
| 130 | + | |
| 131 | + | |
| 132 | + | |
129 | 133 | | |
130 | 134 | | |
131 | 135 | | |
| |||
0 commit comments