File size: 516 Bytes
60616b8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from tsai_gpt.model import GPT
from tsai_gpt.config import Config
from tsai_gpt.tokenizer import Tokenizer
from lightning_utilities.core.imports import RequirementCache
_LIGHTNING_AVAILABLE = RequirementCache("lightning>=2.1.0.dev0")
if not bool(_LIGHTNING_AVAILABLE):
raise ImportError(
"Lit-GPT requires lightning==2.1. Please run:\n"
f" pip uninstall -y lightning; pip install -r requirements.txt\n{str(_LIGHTNING_AVAILABLE)}"
)
__all__ = ["GPT", "Config", "Tokenizer"]
|