LeroyDyer
/

Mixtral_AI_TokenClassification_Project

Token Classification

text-generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

LeroyDyer commited on Apr 11

Commit

12e1e0b

•

1 Parent(s): 09ef976

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -11,7 +11,7 @@ Currently Adding Token Classification to mistral model ;
 Still need some kinks sorting out -on the way - Testing
 ```
 from transformers import AutoModelForTokenClassification, AutoTokenizer
 import torch

 Still need some kinks sorting out -on the way - Testing
 ```
+! pip install flash_attn
 from transformers import AutoModelForTokenClassification, AutoTokenizer
 import torch