English
John6666 commited on
Commit
05c7a14
·
verified ·
1 Parent(s): ff158bc

Upload handler.py

Browse files
Files changed (1) hide show
  1. handler.py +3 -3
handler.py CHANGED
@@ -5,8 +5,8 @@ from diffusers import FluxPipeline, FluxTransformer2DModel, AutoencoderKL, Torch
5
  from PIL.Image import Image
6
  import torch
7
 
8
- import torch._dynamo
9
- torch._dynamo.config.suppress_errors = True
10
 
11
  #from huggingface_inference_toolkit.logging import logger
12
 
@@ -18,7 +18,7 @@ def compile_pipeline(pipe) -> Any:
18
 
19
  class EndpointHandler:
20
  def __init__(self, path=""):
21
- is_compile = True
22
  #repo_id = "camenduru/FLUX.1-dev-diffusers"
23
  repo_id = "NoMoreCopyright/FLUX.1-dev-test"
24
  dtype = torch.bfloat16
 
5
  from PIL.Image import Image
6
  import torch
7
 
8
+ #import torch._dynamo
9
+ #torch._dynamo.config.suppress_errors = True
10
 
11
  #from huggingface_inference_toolkit.logging import logger
12
 
 
18
 
19
  class EndpointHandler:
20
  def __init__(self, path=""):
21
+ is_compile = False
22
  #repo_id = "camenduru/FLUX.1-dev-diffusers"
23
  repo_id = "NoMoreCopyright/FLUX.1-dev-test"
24
  dtype = torch.bfloat16