MCPcopy
hub / github.com/hpcaitech/ColossalAI / test_drafter.py

File test_drafter.py

tests/test_infer/test_drafter.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1import pytest
2import torch
3from transformers import AutoTokenizer, LlamaConfig, LlamaForCausalLM
4

Callers

nothing calls this directly

Calls 3

test_drafterFunction · 0.85
test_spec_decFunction · 0.70
from_pretrainedMethod · 0.45

Tested by

no test coverage detected