From 504f30779f88a6bf4f6ae0e7e0781f501f485667 Mon Sep 17 00:00:00 2001 From: Vixy Date: Thu, 1 Jan 2026 22:23:51 -0600 Subject: [PATCH] Add ENABLE_IP_ADAPTER config flag (default: False) Allows disabling IP-Adapter FaceID entirely for cleaner output quality. Set to True in config.py to re-enable face-locked generation. --- config.py | 1 + worker/generator.py | 5 ++++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/config.py b/config.py index d52aedf..61125c7 100755 --- a/config.py +++ b/config.py @@ -29,6 +29,7 @@ SDXL_REFINER_ID = "stabilityai/stable-diffusion-xl-refiner-1.0" USE_REFINER = False # Set to True to enable refiner (requires more VRAM) # IP-Adapter FaceID settings +ENABLE_IP_ADAPTER = False # Set to True to enable face-locked generation IP_ADAPTER_DIR = Path(os.getenv("DREAMTAIL_IP_ADAPTER", MODELS_DIR / "ip-adapter")) IP_ADAPTER_PATH = IP_ADAPTER_DIR / "ip-adapter-faceid_sdxl.bin" FACE_REFERENCE_DIR = STORAGE_DIR / "faces" # Directory for face reference images diff --git a/worker/generator.py b/worker/generator.py index 2fc3a0c..4d42fa6 100755 --- a/worker/generator.py +++ b/worker/generator.py @@ -244,7 +244,10 @@ class SDXLGenerator: logger.info(f"Parameters: {width}x{height}, steps={num_inference_steps}, guidance={guidance_scale}") # Check if face-locked generation requested - use_face_id = face_image is not None + use_face_id = face_image is not None and config.ENABLE_IP_ADAPTER + + if face_image is not None and not config.ENABLE_IP_ADAPTER: + logger.info("IP-Adapter disabled in config, generating without face lock") if use_face_id: # Initialize IP-Adapter if needed