fix bad import in pytorch notebook

sony · Jun 16, 2024 · 7a0830c · 7a0830c
1 parent 0bc2e61
commit 7a0830c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tutorials/notebooks/mct_features_notebooks/pytorch/example_pytorch_xquant.ipynb b/tutorials/notebooks/mct_features_notebooks/pytorch/example_pytorch_xquant.ipynb
@@ -62,7 +62,7 @@
         "import model_compression_toolkit as mct\n",
         "import numpy as np\n",
         "from functools import partial\n",
-        "from xquant import XQuantConfig\n",
+        "from model_compression_toolkit.xquant import XQuantConfig\n",
         "import torch"
       ]
     },
@@ -152,7 +152,7 @@
         "\n",
         "# Generate the xquant report comparing the float model and the quantized model using the\n",
         "# representative and validation datasets.\n",
-        "from xquant import xquant_report_pytorch_experimental\n",
+        "from model_compression_toolkit.xquant import xquant_report_pytorch_experimental\n",
         "result = xquant_report_pytorch_experimental(\n",
         "            float_model,\n",
         "            quantized_model,\n",