Fix the Phi-3.5 conversion test broken.

PiperOrigin-RevId: 678451578
google-ai-edge · Sep 25, 2024 · c0f0b63 · c0f0b63
1 parent 68904bc
commit c0f0b63
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ai_edge_torch/generative/test/test_model_conversion_large.py b/ai_edge_torch/generative/test/test_model_conversion_large.py
@@ -119,7 +119,7 @@ def test_phi2(self):
   )
   def test_phi3(self):
     config = phi3.get_fake_model_config()
-    pytorch_model = phi3.Phi3(config).eval()
+    pytorch_model = phi3.Phi3_5Mini(config).eval()
     self._test_model(
         config, pytorch_model, "prefill", atol=1e-5, rtol=1e-5
     )