Update webgui.py

webgui.py

@@ -122,7 +122,7 @@ else:
 denoising_unet.load_state_dict(torch.load(config.denoising_unet_path, map_location="cpu"), strict=False)
 
 ## face locator init
-face_locator = FaceLocator(320, conditioning_channels=1, block_out_channels=(16, 32, 96, 256)).to(dtype=weight_dtype, device="cuda")
+face_locator = FaceLocator(320, conditioning_channels=1, block_out_channels=(16, 32, 96, 256)).to("cuda", dtype=weight_dtype)
 face_locator.load_state_dict(torch.load(config.face_locator_path, map_location='cpu'))
 
 ## load audio processor params