Addressed comments

Signed-off-by: Amit Raj <[email protected]>

Author: quic-amitraj

QEfficient/base/modeling_qeff.py

@@ -52,7 +52,7 @@ def __init__(self, model: torch.nn.Module) -> None:
         self.onnx_path: Optional[str] = None
         self.qpc_path: Optional[str] = None
         self.qpc_session: Optional[QAICInferenceSession] = None
-        model = model.to(torch.float32)
+
         # Apply the transformations
         any_transformed = False
         for transform in self._pytorch_transforms: