Update inference_manager.py

inference_manager.py

@@ -205,12 +205,12 @@ class InferenceManager:
             print(ckpt_dir)
             pipe = DiffusionPipeline.from_pretrained(
                 ckpt_dir,
-                vae=vae,
+                #vae=vae,#!TEST! remove vae
                 #unet=unet,
                 torch_dtype=torch.bfloat16,
                 use_safetensors=True,
                 #variant="fp16",
-                custom_pipeline   = "lpw_stable_diffusion_xl",
+                #custom_pipeline   = "lpw_stable_diffusion_xl",#!TEST! remove prompt embedding
             )
             #pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
             clip_skip = cfg.get("clip_skip", 1)