Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -224,7 +224,7 @@ class EndpointHandler:
|
|
| 224 |
)
|
| 225 |
self.model.eval()
|
| 226 |
|
| 227 |
-
|
| 228 |
try:
|
| 229 |
if not hasattr(obj, "forward"):
|
| 230 |
return False
|
|
|
|
| 224 |
)
|
| 225 |
self.model.eval()
|
| 226 |
|
| 227 |
+
def _patch_forward(obj, label="model"):
|
| 228 |
try:
|
| 229 |
if not hasattr(obj, "forward"):
|
| 230 |
return False
|