Update inference_manager.py

inference_manager.py

@@ -515,7 +515,7 @@ class ModelManager:
 
         average_embedding = torch.mean(torch.stack(faceid_all_embeds, dim=0), dim=0)
         average_embedding = average_embedding.to("cuda")
-
+        
         print("start inference...")
         style_selection = ""
         use_negative_prompt = True
@@ -525,6 +525,7 @@ class ModelManager:
         prompt_str = cfg.get("prompt", "{prompt}").replace("{prompt}", p)
         generator = torch.Generator(model.base_model_pipeline.device).manual_seed(seed)
         print(f"generate: p={p}, np={negative_prompt}, steps={steps}, guidance_scale={guidance_scale}, size={width},{height}, seed={seed}")
+        print(f"device: embedding={average_embedding.device}, generator={generator.device}, ip_model={ip_model.device}, pipe={model.base_model_pipeline.device}")
         images = ip_model.generate(
             prompt=prompt_str,
             negative_prompt=negative_prompt,