SYMBOL INDEX (45 symbols across 12 files)

FILE: api.py
  function torch_gc (line 11) | def torch_gc():
  function create_item (line 22) | async def create_item(request: Request):

FILE: cli_demo.py
  function build_prompt (line 19) | def build_prompt(history):
  function signal_handler (line 27) | def signal_handler(signal, frame):
  function main (line 32) | def main():

FILE: evaluation/evaluate_ceval.py
  function build_prompt (line 17) | def build_prompt(text):

FILE: openai_api.py
  function lifespan (line 20) | async def lifespan(app: FastAPI): # collects GPU memory
  class ModelCard (line 37) | class ModelCard(BaseModel):
  class ModelList (line 47) | class ModelList(BaseModel):
  class ChatMessage (line 52) | class ChatMessage(BaseModel):
  class DeltaMessage (line 57) | class DeltaMessage(BaseModel):
  class ChatCompletionRequest (line 62) | class ChatCompletionRequest(BaseModel):
  class ChatCompletionResponseChoice (line 71) | class ChatCompletionResponseChoice(BaseModel):
  class ChatCompletionResponseStreamChoice (line 77) | class ChatCompletionResponseStreamChoice(BaseModel):
  class ChatCompletionResponse (line 83) | class ChatCompletionResponse(BaseModel):
  function list_models (line 91) | async def list_models():
  function create_chat_completion (line 98) | async def create_chat_completion(request: ChatCompletionRequest):
  function predict (line 129) | async def predict(query: str, history: List[List[str]], model_id: str):

FILE: ptuning/arguments.py
  class ModelArguments (line 6) | class ModelArguments:
  class DataTrainingArguments (line 65) | class DataTrainingArguments:
    method __post_init__ (line 212) | def __post_init__(self):

FILE: ptuning/main.py
  function main (line 49) | def main():
  function _mp_fn (line 405) | def _mp_fn(index):

FILE: ptuning/trainer.py
  class PrefixTrainer (line 32) | class PrefixTrainer(Trainer):
    method __init__ (line 33) | def __init__(self, *args, save_changed=False, **kwargs):
    method _save (line 37) | def _save(self, output_dir: Optional[str] = None, state_dict=None):

FILE: ptuning/trainer_seq2seq.py
  class Seq2SeqTrainer (line 30) | class Seq2SeqTrainer(PrefixTrainer):
    method evaluate (line 31) | def evaluate(
    method predict (line 80) | def predict(
    method prediction_step (line 138) | def prediction_step(
    method _pad_tensors_to_max_len (line 231) | def _pad_tensors_to_max_len(self, tensor, max_length):

FILE: ptuning/web_demo.py
  function postprocess (line 28) | def postprocess(self, y):
  function parse_text (line 42) | def parse_text(text):
  function predict (line 75) | def predict(input, chatbot, max_length, top_p, temperature, history, pas...
  function reset_user_input (line 86) | def reset_user_input():
  function reset_state (line 90) | def reset_state():
  function main (line 121) | def main():

FILE: utils.py
  function auto_configure_device_map (line 8) | def auto_configure_device_map(num_gpus: int) -> Dict[str, int]:
  function load_model_on_gpus (line 45) | def load_model_on_gpus(checkpoint_path: Union[str, os.PathLike], num_gpu...

FILE: web_demo.py
  function postprocess (line 16) | def postprocess(self, y):
  function parse_text (line 30) | def parse_text(text):
  function predict (line 63) | def predict(input, chatbot, max_length, top_p, temperature, history, pas...
  function reset_user_input (line 74) | def reset_user_input():
  function reset_state (line 78) | def reset_state():

FILE: web_demo2.py
  function get_model (line 13) | def get_model():