Update inference_manager.py

inference_manager.py

@@ -188,11 +188,10 @@ class InferenceManager:
                 #unet=unet,
                 torch_dtype=torch.bfloat16,
                 use_safetensors=True,
-                sampler=cfg.get("sampler"),
                 #variant="fp16",
                 custom_pipeline   = "lpw_stable_diffusion_xl",
             )
-            
+            pipe.scheduler = cfg.get("sampler")
             clip_skip = cfg.get("clip_skip", 1)
             # Adjust clip skip for XL (assumed not relevant for SD 1.5)
             pipe.text_encoder.config.num_hidden_layers -= (clip_skip - 1)