Update inference_manager.py

inference_manager.py

@@ -203,12 +203,13 @@ class InferenceManager:
             else:
                 vae = AutoencoderTiny.from_pretrained(use_vae, torch_dtype=torch.bfloat16)
             print(ckpt_dir)
-            pipe = DiffusionPipeline.from_pretrained(
-                ckpt_dir,
-                vae=vae,
+            pipe = StableDiffusionXLPipeline.from_pretrained(
+                cfg["model_id"],#ckpt_dir,!TEST! directly use hf path
+                #vae=vae,!TEST!remove vae
                 #unet=unet,
                 torch_dtype=torch.bfloat16,
                 use_safetensors=True,
+                add_watermarker=False,
                 #variant="fp16",
                 #custom_pipeline   = "lpw_stable_diffusion_xl",
             )