make func

Signed-off-by: Mayank Mishra <[email protected]>
IBM · Sep 12, 2024 · e8f5131 · e8f5131
1 parent 94a1b46
commit e8f5131
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/hf_models/single_gpu/model_conversion_test.py b/tests/hf_models/single_gpu/model_conversion_test.py
@@ -94,5 +94,9 @@ def test_granitemoe_model_conversion(self, device: torch.device, attention_head_
         )
 
         self.model_conversion_test(
-            dolomite_config=dolomite_config, model_type="granitemoe", device=device, exact_match=False
+            dolomite_config=dolomite_config,
+            model_type="granitemoe",
+            device=device,
+            exact_match=False,
+            compare_loss=False,
         )