tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: deprecate old types

Browse Source
This commit is contained in:
NanoCode012
2025-02-03 16:03:41 +07:00
parent 053e5fd7d1
commit 077a54d2b1
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/axolotl/core/trainer_builder.py
View File

@@ -25,7 +25,7 @@ import os
import sys import sys
from abc import abstractmethod from abc import abstractmethod
from pathlib import Path from pathlib import Path
from typing import List, Type, Union from typing import Type, Union
import torch import torch
import transformers import transformers
@@ -155,7 +155,7 @@ class TrainerBuilderBase(abc.ABC):
def build(self, total_num_steps): def build(self, total_num_steps):
pass pass
def get_callbacks(self) -> List[TrainerCallback]: def get_callbacks(self) -> list[TrainerCallback]:
callbacks = [] callbacks = []
plugin_manager = PluginManager.get_instance() plugin_manager = PluginManager.get_instance()