* Fix logging import in evaluate.py (#2782) * chore: lint --------- Co-authored-by: Joe Zacaroli <jaz@cyberscience.com> Co-authored-by: Wing Lian <wing@axolotl.ai>
This commit is contained in:
@@ -7,7 +7,6 @@ from pathlib import Path
|
||||
from typing import Dict, Optional
|
||||
|
||||
import torch
|
||||
from accelerate.logging import get_logger
|
||||
from datasets import Dataset
|
||||
from transformers.trainer import Trainer
|
||||
|
||||
@@ -17,6 +16,7 @@ from axolotl.train import (
|
||||
)
|
||||
from axolotl.utils.dict import DictDefault
|
||||
from axolotl.utils.distributed import cleanup_distributed
|
||||
from axolotl.utils.logging import get_logger
|
||||
from axolotl.utils.trainer import setup_trainer
|
||||
|
||||
project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
|
||||
|
||||
Reference in New Issue
Block a user