pass dtype policy

DavidLandup0 · DavidLandup0 · commit 5a6fb2763a9b · 2025-08-05T20:38:52.000+09:00
diff --git a/keras_hub/src/models/smollm3/smollm3_backbone.py b/keras_hub/src/models/smollm3/smollm3_backbone.py
@@ -111,7 +111,8 @@ def __init__(
         #)
         self.rotary_embedding = RotaryEmbedding(
             max_wavelength=rope_theta,
-            scaling_factor=rope_scaling
+            scaling_factor=rope_scaling,
+            dtype=self.dtype_policy
         )
 
         # === Functional Model ===

Original file line number	Diff line number	Diff line change
`@@ -111,7 +111,8 @@ def __init__(`
`111`	`111`	`#)`
`112`	`112`	`self.rotary_embedding = RotaryEmbedding(`
`113`	`113`	`max_wavelength=rope_theta,`
`114`		`- scaling_factor=rope_scaling`
	`114`	`+ scaling_factor=rope_scaling,`
	`115`	`+ dtype=self.dtype_policy`
`115`	`116`	`)`
`116`	`117`
`117`	`118`	`# === Functional Model ===`