Update inference_manager.py

inference_manager.py

@@ -190,10 +190,10 @@ class InferenceManager:
                 ckpt_dir,
                 vae=vae,
                 #unet=unet,
-                torch_dtype=torch.float16,
+                torch_dtype=torch.bfloat16,
                 use_safetensors=True,
                 #variant="fp16",
-                custom_pipeline   = "lpw_stable_diffusion_xl",
+                #custom_pipeline   = "lpw_stable_diffusion_xl",
             )
             #pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
             clip_skip = cfg.get("clip_skip", 1)