From 0e048e434727ff9dd0124da304c5cdab49640c33 Mon Sep 17 00:00:00 2001
From: Adam Kowalski <49454006+akowalsk@users.noreply.github.com>
Date: Tue, 11 Jul 2023 11:35:23 -0500
Subject: [PATCH] Adding logger import to t5_modeling.py

Logger is referenced during the apex importing but is not imported, causing a NameError
---
 .../models/custom_modeling/t5_modeling.py                       | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/server/text_generation_server/models/custom_modeling/t5_modeling.py b/server/text_generation_server/models/custom_modeling/t5_modeling.py
index 1ea82802..d6d864fd 100644
--- a/server/text_generation_server/models/custom_modeling/t5_modeling.py
+++ b/server/text_generation_server/models/custom_modeling/t5_modeling.py
@@ -19,6 +19,8 @@ import math
 import warnings
 from typing import Optional, Tuple, Union
 
+from loguru import logger
+
 import torch
 import torch.distributed
 from torch import nn