Upload handler.py

handler.py

@@ -88,15 +88,11 @@ def load_pipeline_turbo_compile(repo_id: str, dtype: torch.dtype) -> Any:
 
 class EndpointHandler:
     def __init__(self, path=""):
-        repo_id = "camenduru/FLUX.1-dev-diffusers"
+        repo_id = "NoMoreCopyrightOrg/flux-dev-8step" if IS_TURBO else "camenduru/FLUX.1-dev-diffusers"
         #dtype = torch.bfloat16
         dtype = torch.float16 # for older nVidia GPUs
-        if IS_COMPILE:
-            if IS_TURBO: self.pipeline = load_pipeline_turbo_compile(repo_id, dtype)
-            else: self.pipeline = load_pipeline_compile(repo_id, dtype)
-        else:
-            if IS_TURBO: self.pipeline = load_pipeline_turbo(repo_id, dtype)
-            else: self.pipeline = load_pipeline_stable(repo_id, dtype)
+        if IS_COMPILE: load_pipeline_compile(repo_id, dtype)
+        else: self.pipeline = load_pipeline_stable(repo_id, dtype)
 
     def __call__(self, data: Dict[str, Any]) -> Image.Image:
         logger.info(f"Received incoming request with {data=}")