File size: 446 Bytes
29658b2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | import unittest
from specforge.modeling.auto import AutoEagle3DraftModel, LlamaForCausalLMEagle3
class TestAutoModelForCausalLM(unittest.TestCase):
def test_automodel(self):
"""init"""
model = AutoEagle3DraftModel.from_pretrained(
"jamesliu1/sglang-EAGLE3-Llama-3.1-Instruct-8B"
)
self.assertIsInstance(model, LlamaForCausalLMEagle3)
if __name__ == "__main__":
unittest.main(verbosity=2)
|