updated changes of mistral_inference in transformer class

mistralai · Jul 23, 2024 · f38fb25 · f38fb25
1 parent 5b8adb5
commit f38fb25
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tutorials/mistral_finetune_7b.ipynb b/tutorials/mistral_finetune_7b.ipynb
@@ -1840,7 +1840,7 @@
     }
    ],
    "source": [
-    "from mistral_inference.model import Transformer\n",
+    "from mistral_inference.transformer import Transformer\n",
     "from mistral_inference.generate import generate\n",
     "\n",
     "from mistral_common.tokens.tokenizers.mistral import MistralTokenizer\n",