| from .config import * | |
| from .model import * | |
| # from .model.tokenizer import * | |
| from .configuration_olmo import OLMoConfig | |
| from .modeling_olmo import OLMoForCausalLM | |
| from .modeling_olmo import OLMoForSequenceCLS | |
| from .tokenization_olmo_fast import OLMoTokenizerFast | |
| def check_install(cuda: bool = False): | |
| import torch | |
| from .version import VERSION | |
| if cuda: | |
| assert torch.cuda.is_available(), "CUDA is not available!" | |
| print("CUDA available") | |
| print(f"OLMo v{VERSION} installed") | |