python inference/genefacepp_infer.py

[newworld1020]

G:\GeneFacePlusPlus.\modules\radnerfs\utils.py:38: FutureWarning: torch.cuda.amp.custom_fwd(args...) is deprecated. Please use torch.amp.custom_fwd(args..., device_type='cuda') instead.
@custom_fwd(cast_inputs=torch.float32, device_type='cuda')#@custom_fwd(cast_inputs=torch.float32) # cast to float32
Traceback (most recent call last):
File "G:\GeneFacePlusPlus\inference\genefacepp_infer.py", line 35, in
from tasks.radnerfs.dataset_utils import RADNeRFDataset, get_boundary_mask, dilate_boundary_mask, get_lf_boundary_mask
File "G:\GeneFacePlusPlus.\tasks\radnerfs\dataset_utils.py", line 16, in
from modules.radnerfs.utils import get_audio_features, get_rays, get_bg_coords, convert_poses, nerf_matrix_to_ngp
File "G:\GeneFacePlusPlus.\modules\radnerfs\utils.py", line 36, in
class _trunc_exp(Function):
File "G:\GeneFacePlusPlus.\modules\radnerfs\utils.py", line 38, in _trunc_exp
@custom_fwd(cast_inputs=torch.float32, device_type='cuda')#@custom_fwd(cast_inputs=torch.float32) # cast to float32
File "D:\ANACONDA\envs\geneface\lib\site-packages\typing_extensions.py", line 2636, in wrapper
return arg(*args, **kwargs)
TypeError: custom_fwd() got an unexpected keyword argument 'device_type'