Views
No views yet
llm2vec library.1import torch
2from llm2vec import LLM2Vec
3
4model_id = "LSX-UniWue/LLaMmlein2Vec_7B"
5l2v = LLM2Vec.from_pretrained(
6 model_id,
7 device_map="cuda" if torch.cuda.is_available() else "cpu",
8 torch_dtype=torch.bfloat16,
9)