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

Use default scheduler when IP-Adapter disabled

Browse Source 
DDIM scheduler was always active, causing softer output. Now only
uses DDIM when ENABLE_IP_ADAPTER=True, otherwise uses model's
default scheduler for best quality.
This commit is contained in:
Vixy
2026-01-01 22:31:39 -06:00
parent 504f30779f
commit 8e0f4bf20f
1 changed files with 31 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
worker/generator.py
View File

@@ -60,25 +60,37 @@ class SDXLGenerator:
try:
dtype = torch.float16 if config.USE_FP16 else torch.float32
# Use DDIM scheduler for IP-Adapter compatibility
noise_scheduler = DDIMScheduler(
num_train_timesteps=1000,
beta_start=0.00085,
beta_end=0.012,
beta_schedule="scaled_linear",
clip_sample=False,
set_alpha_to_one=False,
steps_offset=1,
)
# Only use DDIM scheduler when IP-Adapter is enabled (it requires DDIM)
# Otherwise let the model use its default scheduler for best quality
if config.ENABLE_IP_ADAPTER:
noise_scheduler = DDIMScheduler(
num_train_timesteps=1000,
beta_start=0.00085,
beta_end=0.012,
beta_schedule="scaled_linear",
clip_sample=False,
set_alpha_to_one=False,
steps_offset=1,
)
logger.info("Using DDIM scheduler for IP-Adapter compatibility")
self.pipeline = StableDiffusionXLPipeline.from_pretrained(
config.SDXL_MODEL_ID,
torch_dtype=dtype,
scheduler=noise_scheduler,
use_safetensors=True,
cache_dir=str(config.MODELS_DIR),
add_watermarker=False,
)
self.pipeline = StableDiffusionXLPipeline.from_pretrained(
config.SDXL_MODEL_ID,
torch_dtype=dtype,
scheduler=noise_scheduler,
use_safetensors=True,
cache_dir=str(config.MODELS_DIR),
add_watermarker=False,
)
else:
# Use model's default scheduler for best quality
self.pipeline = StableDiffusionXLPipeline.from_pretrained(
config.SDXL_MODEL_ID,
torch_dtype=dtype,
use_safetensors=True,
cache_dir=str(config.MODELS_DIR),
add_watermarker=False,
)
# Move to device
self.pipeline = self.pipeline.to(self.device)