From af4803000e8f2d085ccfdc390bcd7518228bac6f Mon Sep 17 00:00:00 2001
From: Ella Charlaix <ella@huggingface.co>
Date: Thu, 22 Feb 2024 11:59:04 +0100
Subject: [PATCH] test

---
 optimum/exporters/openvino/__main__.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/optimum/exporters/openvino/__main__.py b/optimum/exporters/openvino/__main__.py
index d0932d965c..417edac1c0 100644
--- a/optimum/exporters/openvino/__main__.py
+++ b/optimum/exporters/openvino/__main__.py
@@ -224,6 +224,7 @@ class StoreAttr(object):
             return model
 
         GPTQQuantizer.post_init_model = post_init_model
+    import torch
 
     model = TasksManager.get_model_from_task(
         task,
@@ -238,7 +239,7 @@ class StoreAttr(object):
         framework=framework,
         device=device,
         library_name=library_name,
-        torch_dype=torch.float32,
+        torch_dtype=torch.float32,
         **loading_kwargs,
     )