test

huggingface · Feb 22, 2024 · af48030 · af48030
1 parent 1aef938
commit af48030
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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,
     )