Rank 0-only logging (#2608)
Co-authored-by: Wing Lian <wing@axolotl.ai>
This commit is contained in:
@@ -15,12 +15,13 @@
|
||||
"""
|
||||
Module for handling LIGER input arguments.
|
||||
"""
|
||||
import logging
|
||||
from typing import Optional
|
||||
|
||||
from pydantic import BaseModel, model_validator
|
||||
|
||||
LOG = logging.getLogger("axolotl.integrations.liger.args")
|
||||
from axolotl.utils.logging import get_logger
|
||||
|
||||
LOG = get_logger(__name__)
|
||||
|
||||
|
||||
class LigerArgs(BaseModel):
|
||||
|
||||
Reference in New Issue
Block a user