Add DP research model (#2396)

sachinprasadhs · buildwithsuhana · abheesht17 · web-flow · commit f4b648d58e18 · 2025-09-12T08:30:37.000-07:00
* Fixes (#2395) * Adding fixes * Adding fixes * Adding Fixes * patch release * Disable DINO quantisation checks (#2397) --------- Co-authored-by: Suhana <suhanaaa@google.com> Co-authored-by: Abheesht <sharmabhee@gmail.com>
diff --git a/keras_hub/src/models/dinov2/dinov2_backbone_test.py b/keras_hub/src/models/dinov2/dinov2_backbone_test.py
@@ -35,6 +35,7 @@ def test_backbone_basics(self):
             init_kwargs=self.init_kwargs,
             input_data=self.input_data,
             expected_output_shape=(2, sequence_length, hidden_dim),
+            run_quantization_check=False,
         )
 
     @pytest.mark.large
@@ -126,6 +127,7 @@ def test_backbone_basics(self):
             init_kwargs=self.init_kwargs,
             input_data=self.input_data,
             expected_output_shape=(2, sequence_length, hidden_dim),
+            run_quantization_check=False,
         )
 
     @pytest.mark.large
diff --git a/keras_hub/src/models/gemma/gemma_presets.py b/keras_hub/src/models/gemma/gemma_presets.py
@@ -198,4 +198,12 @@
         },
         "kaggle_handle": "kaggle://google/shieldgemma/keras/shieldgemma_27b_en/2",
     },
+    "vault_gemma_1b_en": {
+        "metadata": {
+            "description": "1 billion parameter, 26-layer, VaultGemma model.",
+            "params": 1038741120,
+            "path": "gemma",
+        },
+        "kaggle_handle": "kaggle://keras/vaultgemma/keras/vault_gemma_1b_en/2",
+    },
 }
diff --git a/keras_hub/src/version.py b/keras_hub/src/version.py
@@ -1,7 +1,7 @@
 from keras_hub.src.api_export import keras_hub_export
 
 # Unique source of truth for the version number.
-__version__ = "0.22.1"
+__version__ = "0.22.2"
 
 
 @keras_hub_export("keras_hub.version")

Original file line number	Diff line number	Diff line change
`@@ -35,6 +35,7 @@ def test_backbone_basics(self):`
`35`	`35`	`init_kwargs=self.init_kwargs,`
`36`	`36`	`input_data=self.input_data,`
`37`	`37`	`expected_output_shape=(2, sequence_length, hidden_dim),`
	`38`	`+ run_quantization_check=False,`
`38`	`39`	`)`
`39`	`40`
`40`	`41`	`@pytest.mark.large`
`@@ -126,6 +127,7 @@ def test_backbone_basics(self):`
`126`	`127`	`init_kwargs=self.init_kwargs,`
`127`	`128`	`input_data=self.input_data,`
`128`	`129`	`expected_output_shape=(2, sequence_length, hidden_dim),`
	`130`	`+ run_quantization_check=False,`
`129`	`131`	`)`
`130`	`132`
`131`	`133`	`@pytest.mark.large`