Update models/embedding_manager.py

models/embedding_manager.py

@@ -123,7 +123,7 @@ class EmbeddingManagerId_adain(nn.Module):
 
 
     def load(self, ckpt_path):
-        ckpt = torch.load(ckpt_path, map_location='cuda')
+        ckpt = torch.load(ckpt_path, map_location='cuda', weights_only=False)
         if ckpt.get("name_projection_layer") is not None:
             self.name_projection_layer = ckpt.get("name_projection_layer").float()