File chat_sat.py

SwissArmyTransformer/examples/llama/chat_sat.py:None–None

import torch
from sat import AutoModel
from transformers import LlamaTokenizer
from sat.model.mixins import CachedAutoregressiveMixin