Skip to content

Commit

Permalink
Merge pull request #83 from bhargavyagnik/main
Browse files Browse the repository at this point in the history
updated changes of mistral_inference in transformer class
  • Loading branch information
pandora-s-git authored Jul 29, 2024
2 parents d137146 + f38fb25 commit a9b3c98
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tutorials/mistral_finetune_7b.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down

0 comments on commit a9b3c98

Please sign in to comment.