merge-hf-and-lora-to-hf.py

This commit is contained in:
FNsi 2023-05-22 13:08:59 +08:00 committed by GitHub
parent 651f50f6ca
commit b6a30489a7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1,4 +1,3 @@
import json
import torch
import argparse
from transformers import LlamaTokenizer, LlamaForCausalLM