diff --git a/src/llamafactory/__init__.py b/src/llamafactory/__init__.py index bb98bf5c..18f04b0a 100644 --- a/src/llamafactory/__init__.py +++ b/src/llamafactory/__init__.py @@ -31,6 +31,12 @@ Dependency graph: transformers>=4.41.2,<=4.44.3 packing: transformers>=4.41.2,<=4.44.3 + +Disable version checking: DISABLE_VERSION_CHECK=1 +Enable VRAM recording: RECORD_VRAM=1 +Force check imports: FORCE_CHECK_IMPORTS=1 +Force using torchrun: FORCE_TORCHRUN=1 +Use modelscope: USE_MODELSCOPE_HUB=1 """ from .extras.env import VERSION diff --git a/src/llamafactory/train/callbacks.py b/src/llamafactory/train/callbacks.py index 94e1541c..dc1dd0ad 100644 --- a/src/llamafactory/train/callbacks.py +++ b/src/llamafactory/train/callbacks.py @@ -318,7 +318,7 @@ class LogCallback(TrainerCallback): if self.webui_mode and all(key in logs for key in ["loss", "learning_rate", "epoch"]): logger.info( "{{'loss': {:.4f}, 'learning_rate': {:2.4e}, 'epoch': {:.2f}, 'throughput': {}}}".format( - logs["loss"], logs["learning_rate"], logs["epoch"], logs.get("throughput") + logs["loss"], logs["learning_rate"], logs["epoch"], logs.get("throughput", "N/A") ) )