Fix a typo in model name of OpenELM

ai-edge-bot · copybara-github · commit f2045c466f25 · 2024-12-10T11:45:20.000-08:00
PiperOrigin-RevId: 704791093
diff --git a/ai_edge_torch/generative/test/test_model_conversion_large.py b/ai_edge_torch/generative/test/test_model_conversion_large.py
@@ -151,7 +151,7 @@ def test_smollm(self):
   )
   def test_openelm(self):
     config = openelm.get_fake_model_config()
-    pytorch_model = openelm.OpenElm(config).eval()
+    pytorch_model = openelm.OpenELM(config).eval()
     self._test_model(config, pytorch_model, "prefill", atol=1e-4, rtol=1e-5)
 
   @googletest.skipIf(

Original file line number	Diff line number	Diff line change
`@@ -151,7 +151,7 @@ def test_smollm(self):`
`151`	`151`	`)`
`152`	`152`	`def test_openelm(self):`
`153`	`153`	`config = openelm.get_fake_model_config()`
`154`		`- pytorch_model = openelm.OpenElm(config).eval()`
	`154`	`+ pytorch_model = openelm.OpenELM(config).eval()`
`155`	`155`	`self._test_model(config, pytorch_model, "prefill", atol=1e-4, rtol=1e-5)`
`156`	`156`
`157`	`157`	`@googletest.skipIf(`