zmactep

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import json
import os
import argparse
import os.path as osp
from typing import Union
import torch
from transformers import GenerationConfig, LlamaForCausalLM, LlamaTokenizer