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