fix

Signed-off-by: Hemil Desai <hemild@nvidia.com>

Author: hemildesai

tests/unit_tests/models/deepseek_v3/test_dsv3_layers.py

@@ -201,6 +201,7 @@ def create_mock_config(self, **overrides):
         config.hidden_size = 4096
         config.rope_scaling = None
         config.max_position_embeddings = 4096
+        config.rms_norm_eps = 1e-6
 
         # Apply overrides
         for key, value in overrides.items():
@@ -358,6 +359,7 @@ def create_mock_config(self, **overrides):
         config.hidden_size = 1024
         config.rope_scaling = None
         config.max_position_embeddings = 4096
+        config.rms_norm_eps = 1e-6
 
         for key, value in overrides.items():
             setattr(config, key, value)
@@ -480,6 +482,7 @@ def create_mock_config(self, **overrides):
         config.hidden_size = 1024
         config.rope_scaling = None
         config.max_position_embeddings = 4096
+        config.rms_norm_eps = 1e-6
 
         for key, value in overrides.items():
             setattr(config, key, value)