fix(server): Adding logger import to t5_modeling.py (#585)
Logger is referenced during the apex importing but is not imported, causing a NameError
This commit is contained in:
parent
db4efbf4bc
commit
7f9072228a
|
@ -19,6 +19,8 @@ import math
|
||||||
import warnings
|
import warnings
|
||||||
from typing import Optional, Tuple, Union
|
from typing import Optional, Tuple, Union
|
||||||
|
|
||||||
|
from loguru import logger
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
import torch.distributed
|
import torch.distributed
|
||||||
from torch import nn
|
from torch import nn
|
||||||
|
|
Loading…
Reference in New Issue