vixy/dreamtail
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add ENABLE_IP_ADAPTER config flag (default: False)

Browse Source 
Allows disabling IP-Adapter FaceID entirely for cleaner output quality.
Set to True in config.py to re-enable face-locked generation.
This commit is contained in:
Vixy
2026-01-01 22:23:51 -06:00
parent f5e37bc5b2
commit 504f30779f
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
config.py
View File

@@ -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) USE_REFINER = False # Set to True to enable refiner (requires more VRAM)
# IP-Adapter FaceID settings # 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_DIR = Path(os.getenv("DREAMTAIL_IP_ADAPTER", MODELS_DIR / "ip-adapter"))
IP_ADAPTER_PATH = IP_ADAPTER_DIR / "ip-adapter-faceid_sdxl.bin" IP_ADAPTER_PATH = IP_ADAPTER_DIR / "ip-adapter-faceid_sdxl.bin"
FACE_REFERENCE_DIR = STORAGE_DIR / "faces" # Directory for face reference images FACE_REFERENCE_DIR = STORAGE_DIR / "faces" # Directory for face reference images

5
worker/generator.py
View File

@@ -244,7 +244,10 @@ class SDXLGenerator:
logger.info(f"Parameters: {width}x{height}, steps={num_inference_steps}, guidance={guidance_scale}") logger.info(f"Parameters: {width}x{height}, steps={num_inference_steps}, guidance={guidance_scale}")
# Check if face-locked generation requested # 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: if use_face_id:
# Initialize IP-Adapter if needed # Initialize IP-Adapter if needed