Update inference_manager.py

inference_manager.py

@@ -514,6 +514,7 @@ class ModelManager:
             faceid_all_embeds.append(faceid_embed)
 
         average_embedding = torch.mean(torch.stack(faceid_all_embeds, dim=0), dim=0)
+        average_embedding = average_embedding.to("cuda")
 
         print("start inference...")
         style_selection = ""