huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-10 20:04:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adding logger import to t5_modeling.py

Browse Source 
Logger is referenced during the apex importing but is not imported, causing a NameError
This commit is contained in:
Adam Kowalski 2023-07-11 11:35:23 -05:00 committed by GitHub
parent b4024edd45
commit 0e048e4347
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