Skip to content

Commit

Permalink
Revert "test (ex/llm): Removed onnx dependency when collecting tests."
Browse files Browse the repository at this point in the history
This reverts commit d4509c6.
  • Loading branch information
nickfraser committed Aug 23, 2024
1 parent e7cf788 commit 501160d
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions tests/brevitas_examples/test_llm.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
import shutil

import numpy as np
import onnx
import pytest
import torch

Expand Down Expand Up @@ -331,9 +332,6 @@ def onnx_export_args(default_run_args, request):
@jit_disabled_for_export()
@requires_pt_ge('2.2')
def test_small_models_onnx_export(caplog, onnx_export_args):
import onnx

from brevitas_examples.llm.main import main
caplog.set_level(logging.INFO)
args = onnx_export_args
float_ppl, quant_ppl, model = main(args)
Expand Down

0 comments on commit 501160d

Please sign in to comment.