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

missing transformers import

Browse Source
This commit is contained in:
Wing Lian
2023-08-28 05:43:19 -04:00
parent a6c9223114
commit 606846e0a5
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/axolotl/utils/trainer.py
View File

@@ -12,6 +12,7 @@ from typing import Optional, Union
import numpy as np import numpy as np
import torch.cuda import torch.cuda
import transformers
from datasets import Dataset, set_caching_enabled from datasets import Dataset, set_caching_enabled
from torch.optim.lr_scheduler import OneCycleLR from torch.optim.lr_scheduler import OneCycleLR
from torch.utils.data import ( from torch.utils.data import (