Marlon Wiprud commited on
Commit ·
fc124cf
1
Parent(s): c792b0d
log
Browse files- handler.py +6 -0
handler.py
CHANGED
|
@@ -10,6 +10,12 @@ from accelerate import (
|
|
| 10 |
load_checkpoint_and_dispatch,
|
| 11 |
)
|
| 12 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
|
| 14 |
class EndpointHandler:
|
| 15 |
def __init__(self, path=""):
|
|
|
|
| 10 |
load_checkpoint_and_dispatch,
|
| 11 |
)
|
| 12 |
|
| 13 |
+
import logging
|
| 14 |
+
from transformers import logging as hf_logging
|
| 15 |
+
|
| 16 |
+
logging.basicConfig(level=logging.INFO)
|
| 17 |
+
hf_logging.set_verbosity_debug()
|
| 18 |
+
|
| 19 |
|
| 20 |
class EndpointHandler:
|
| 21 |
def __init__(self, path=""):
|