add ZeroGPU support

inference_utils.py

@@ -67,9 +67,9 @@ def init_pipeline():
     id_encoder        = ControlNetModel.from_unet(Unet)
     pose_encoder      = ControlNetModel.from_unet(Unet)
     makeup_encoder    = detail_encoder(Unet, "openai/clip-vit-large-patch14", device=device, dtype=torch.float16)
-    id_state_dict     = torch.load(id_encoder_path, map_location=device)
-    pose_state_dict   = torch.load(pose_encoder_path, map_location=device)
-    makeup_state_dict = torch.load(makeup_encoder_path, map_location=device)
+    id_state_dict     = torch.load(id_encoder_path, map_location=torch.device('cpu'))
+    pose_state_dict   = torch.load(pose_encoder_path, map_location=torch.device('cpu'))
+    makeup_state_dict = torch.load(makeup_encoder_path, map_location=torch.device('cpu'))
     id_encoder.load_state_dict(id_state_dict, strict=False)
     pose_encoder.load_state_dict(pose_state_dict, strict=False)
     makeup_encoder.load_state_dict(makeup_state_dict, strict=False)