Addressed comments

Signed-off-by: Amit Raj <quic_amitraj@quicinc.com>

Author: quic-amitraj

QEfficient/base/modeling_qeff.py

@@ -18,7 +18,6 @@
 
 import onnx
 import torch
-import torch.nn as nn
 
 from QEfficient.base.onnx_transforms import OnnxTransform
 from QEfficient.base.pytorch_transforms import PytorchTransform
@@ -121,7 +120,6 @@ def _export(
         export_kwargs: Optional[Dict[str, any]] = None,
         onnx_transform_kwargs: Optional[Dict[str, any]] = None,
         export_dir: Optional[str] = None,
-        model: nn.Module = None,
     ) -> str:
         """
         Export the Pytorch model to ONNX.