Upload handler.py
Browse files- 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 =
|
| 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
|