Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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:
Adam Kowalski 2023-07-12 03:40:32 -05:00 committed by GitHub
parent db4efbf4bc
commit 7f9072228a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/text_generation_server/models/custom_modeling/t5_modeling.py
View File

@ -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