yibinlei
/

LENS-d4000

Feature Extraction

sentence-similarity

text-generation-inference

text-embeddings-inference

Inference Endpoints

Model card Files Files and versions Community

Yibin Lei commited on 13 days ago

Commit

e473b33

·

1 Parent(s): d5ab748

Update readme

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -8683,7 +8683,7 @@ docs = ["Since 2007, the RBA's outstanding reputation has been affected by the '
         "Today's oilskins (or oilies) typically come in two parts, jackets and trousers. Oilskin jackets are generally similar to common rubberized waterproofs."]
 # Load the model and tokenizer
-model = MistralBiForCausalLM.from_pretrained("yibinlei/LENS-d4000")
 model.lm_head = torch.load('lm_head.pth')
 tokenizer = AutoTokenizer.from_pretrained("yibinlei/LENS-d4000")

         "Today's oilskins (or oilies) typically come in two parts, jackets and trousers. Oilskin jackets are generally similar to common rubberized waterproofs."]
 # Load the model and tokenizer
+model = MistralBiForCausalLM.from_pretrained("yibinlei/LENS-d4000", ignore_mismatched_sizes=True)
 model.lm_head = torch.load('lm_head.pth')
 tokenizer = AutoTokenizer.from_pretrained("yibinlei/LENS-d4000")