fee3c13bb5
* only configure logging on cli to play nicely with colab * allow reloading the config on the fly from a dict * make sure to use dict for yaml * reuse existing function for load * make cli args optional * mps fix and respect max_steps
9 lines
173 B
Python
9 lines
173 B
Python
"""Axolotl CLI module initialization."""
|
|
|
|
import os
|
|
|
|
from axolotl.logging_config import configure_logging
|
|
|
|
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
|
|
configure_logging()
|