Can't use .half() with new GFPGAN clean

[marcelodiaz558]

Hi! For the previous paper version, I was used to using FP16 weights with the model, which helped to have faster processing, now very weird results are generated if I apply model.half() and I use HalfTensors as the model input.

This is what I am doing:

For loading the model:

gfpgan = gfpgan.half()

For the model inputs:

cropped_faces_t = cropped_faces_t.half()

When I remove the above lines of codes everything works fine, but inference time is 44% bigger compared to the paper's version with half weights (looks like that quality isn't that good either, but that's another story). With .half() inference time is 20% bigger, but no result is generated.

These are the results I get:

[xinntao]

We have also noticed this problem.

It is because this line:

GFPGAN/gfpgan/archs/stylegan2_clean_arch.py

Line 21 in d7cb9f7

return x * torch.rsqrt(torch.mean(x**2, dim=1, keepdim=True) + 1e-8)

this operation will have a numerical error when using half.

We may remove this line and fine-tune the model, so that the new model can also use the half inference.

[zhoumenghan]

We have also noticed this problem.

It is because this line:

GFPGAN/gfpgan/archs/stylegan2_clean_arch.py

Line 21 in d7cb9f7

return x * torch.rsqrt(torch.mean(x**2, dim=1, keepdim=True) + 1e-8)

this operation will have a numerical error when using half.

We may remove this line and fine-tune the model, so that the new model can also use the half inference.

any update about the half inference？