remove unnecessary import

Signed-off-by: Xavier Dupre <[email protected]>
microsoft · Jun 17, 2024 · 15799d5 · 15799d5
1 parent f50e5d6
commit 15799d5
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/onnxscript/tools/transformers_models/mistral.py b/onnxscript/tools/transformers_models/mistral.py
@@ -24,8 +24,6 @@ def _prepare_config_and_inputs(
     use_token_type_ids: bool = False,
     use_labels: bool = False,
 ) -> tuple[Any, ...]:
-    import torch
-
     input_ids = onnxscript.tools.transformers_models.ids_tensor(
         [batch_size, seq_length], vocab_size
     )
@@ -87,7 +85,6 @@ def get_mistral_model(
     <https://huggingface.co/docs/transformers/main/en/model_doc/mistral#transformers.MistralConfig>`_.
     The parameters are chosen for a unit test configuration.
     """
-    import torch
     from transformers import MistralConfig
     from transformers.models.mistral.modeling_mistral import MistralModel