2024-06-26 09:33:45 -07:00
|
|
|
import argparse
|
2024-01-08 04:37:50 +00:00
|
|
|
import code
|
|
|
|
|
|
2024-06-26 09:33:45 -07:00
|
|
|
from sglang.srt.hf_transformers_utils import get_tokenizer
|
2024-01-08 04:37:50 +00:00
|
|
|
|
2024-06-26 09:33:45 -07:00
|
|
|
if __name__ == "__main__":
|
|
|
|
|
parser = argparse.ArgumentParser()
|
2024-07-18 04:55:39 +10:00
|
|
|
parser.add_argument(
|
|
|
|
|
"--name", type=str, default="meta-llama/Meta-Llama-3-8B-Instruct"
|
|
|
|
|
)
|
2024-06-26 09:33:45 -07:00
|
|
|
args = parser.parse_args()
|
|
|
|
|
|
|
|
|
|
t = get_tokenizer(args.name)
|
2024-07-18 04:55:39 +10:00
|
|
|
code.interact(local=locals())
|