NoMoreCopyright
/

FLUX.1-dev-test

Model card Files Files and versions Community

NoMoreCopyright commited on Mar 4

Commit

0b0f70b

·

verified ·

1 Parent(s): ce2c3be

Upload handler.py

Files changed (1) hide show

handler.py +2 -1

handler.py CHANGED Viewed

@@ -18,7 +18,8 @@ def compile_pipeline(pipe):
 class EndpointHandler:
     def __init__(self, **kwargs: Any) -> None:  # type: ignore
         is_compile = False
-        repo_id = "camenduru/FLUX.1-dev-diffusers"
         dtype = torch.bfloat16
         quantization_config = TorchAoConfig("int4dq")
         vae = AutoencoderKL.from_pretrained(repo_id, subfolder="vae", torch_dtype=dtype)

 class EndpointHandler:
     def __init__(self, **kwargs: Any) -> None:  # type: ignore
         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)