test

Author: echarlaix

optimum/exporters/openvino/__main__.py

@@ -223,7 +223,6 @@ class StoreAttr(object):
             return model
 
         GPTQQuantizer.post_init_model = post_init_model
-    import torch
 
     model = TasksManager.get_model_from_task(
         task,
@@ -238,7 +237,6 @@ class StoreAttr(object):
         framework=framework,
         device=device,
         library_name=library_name,
-        # torch_dtype=torch.float32,
         **loading_kwargs,
     )
 

setup.py

@@ -50,7 +50,8 @@
         "onnxruntime",
     ],
     "openvino-tokenizers": ["openvino-tokenizers[transformers]"],
-    "nncf": ["nncf>=2.8.1"],
+    # "nncf": ["nncf>=2.8.1"],
+    "nncf": ["nncf @ git+https://github.com/openvinotoolkit/nncf.git"],
     "ipex": ["intel-extension-for-pytorch", "onnx"],
     "diffusers": ["diffusers"],
     "quality": QUALITY_REQUIRE,