NoMoreCopyrightOrg
/

flux-test

Model card Files Files and versions

John6666 commited on Mar 4, 2025

Commit

3fcc2ae

·

verified ·

1 Parent(s): ba3e6ba

Upload handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -19,8 +19,8 @@ def compile_pipeline(pipe) -> Any:
 class EndpointHandler:
     def __init__(self, path=""):
         is_compile = False
-        #repo_id = "camenduru/FLUX.1-dev-diffusers"
-        repo_id = "NoMoreCopyright/FLUX.1-dev-test"
         dtype = torch.bfloat16
         quantization_config = TorchAoConfig("int4dq")
         vae = AutoencoderKL.from_pretrained(repo_id, subfolder="vae", torch_dtype=dtype)

 class EndpointHandler:
     def __init__(self, path=""):
         is_compile = False
+        repo_id = "camenduru/FLUX.1-dev-diffusers"
+        #repo_id = "NoMoreCopyright/FLUX.1-dev-test"
         dtype = torch.bfloat16
         quantization_config = TorchAoConfig("int4dq")
         vae = AutoencoderKL.from_pretrained(repo_id, subfolder="vae", torch_dtype=dtype)