Xenova

/

llama2.c-stories110M

like 5

@dsikka

from transformers import AutoModelForCausalLM
model_loaded = AutoModelForCausalLM.from_pretrained("Xenova/llama2.c-stories110M")
state_dict = model_loaded.state_dict()
for k, v in state_dict.items():
    print(k, v.device)