Skip to content

Commit

Permalink
Fix merge conflict
Browse files Browse the repository at this point in the history
  • Loading branch information
gramalingam committed Jan 7, 2025
1 parent e0f29e2 commit 41aa177
Showing 1 changed file with 0 additions and 8 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -6,21 +6,13 @@

import onnxscript.optimizer
from onnxscript.rewriter.onnxruntime.xformers import fuse_cos_sin_cache, fuse_rotary_embedding
<<<<<<< HEAD
from onnxscript.rewriter.onnxruntime.xformers._smollm_1 import TestData
=======
from onnxscript.rewriter.onnxruntime.xformers._smollm_1layer import _SmollmTestData
>>>>>>> main
from onnxscript.rewriter.onnxruntime.xformers._test_utils import assert_allclose, ort_run


class TestCosSinCacheTransform(unittest.TestCase):
def test_smollm(self):
<<<<<<< HEAD
smollm_test = TestData()
=======
smollm_test = _SmollmTestData()
>>>>>>> main
model = smollm_test.get_onnx_model()
onnxscript.optimizer.optimize(model)
inputs = smollm_test.get_ort_inputs()
Expand Down

0 comments on commit 41aa177

Please sign in to comment.