Upload handler.py

handler.py

@@ -8,7 +8,7 @@ import torch
 import torch._dynamo
 torch._dynamo.config.suppress_errors = True
 
-from huggingface_inference_toolkit.logging import logger
+#from huggingface_inference_toolkit.logging import logger
 
 def compile_pipeline(pipe) -> Any:
     pipe.transformer.fuse_qkv_projections()
@@ -31,7 +31,7 @@ class EndpointHandler:
 
     #@torch.inference_mode()
     def __call__(self, data: Dict[str, Any]) -> Image:
-        logger.info(f"Received incoming request with {data=}")
+        #logger.info(f"Received incoming request with {data=}")
 
         if "inputs" in data and isinstance(data["inputs"], str):
             prompt = data.pop("inputs")