THUDM / SwissArmyTransformer

SwissArmyTransformer is a flexible and powerful library to develop your own Transformer variants.

Home Page:https://THUDM.github.io/SwissArmyTransformer

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to load and initialize llama2 models downloaded from Huggingface

microhu opened this issue · comments

Does the sat lib support. llama2? I found the parameters are not initilalized when using lamma class to load the vicuna-7b-1.5 model by using from_pretrained func.
Could you kiindly give an. example of using sat to load and init the official llama2 models ?

We support llama2 in sat. But I haven't uploaded the weights to cloud. You can use the scripts here to transform the model downloaded from huggingface to sat format: https://github.com/THUDM/SwissArmyTransformer/tree/main/examples/llama

We support llama2 in sat. But I haven't uploaded the weights to cloud. You can use the scripts here to transform the model downloaded from huggingface to sat format: https://github.com/THUDM/SwissArmyTransformer/tree/main/examples/llama

got, thanks. I will try