fireballoon
/

baichuan-llama-7b

Text Generation

text-generation-inference

Model card Files Files and versions Community

fireballoon commited on Jun 16, 2023

Commit

7502c3a

•

1 Parent(s): 3cdc5c4

Update convert_baichuan_to_llama.py

Files changed (1) hide show

convert_baichuan_to_llama.py +1 -1

convert_baichuan_to_llama.py CHANGED Viewed

@@ -11,4 +11,4 @@ for key in baichuan:
         llama[key.replace('W_pack', 'v_proj')] = baichuan[key][4096 * 2:]
     else:
         llama[key] = baichuan[key]
-torch.save(baichuan, "pytorch_model.bin")

         llama[key.replace('W_pack', 'v_proj')] = baichuan[key][4096 * 2:]
     else:
         llama[key] = baichuan[key]
+torch.save(llama, "pytorch_model.bin")