Я получил значения скрытого состояния последнего слоя следующим образом. :
Код: Выделить всё
from transformers import LlamaModel, LlamaTokenizer, LlamaForCausalLM
tokenizer = LlamaTokenizer.from_pretrained(path_to_llama2)
model = LlamaModel.from_pretrained(path_to_llama2)
model_ = LlamaForCausalLM.from_pretrained(path_to_llama2)
tokenizer.pad_token = tokenizer.eos_token
inputs = tokenizer(prompt, return_tensors='pt')
with torch.no_grad():
outputs = model(**inputs, output_attentions=True, output_hidden_states=True)
hidden_states = outputs.hidden_states[-1] # Last layer hidden states
Подробнее здесь: https://stackoverflow.com/questions/784 ... ate-values