Error when running inference for dataset #11

Hi! I was trying to run the inference script for dataset and I obtained the following error.
Traceback (most recent call last):
File "inference_for_dataset.py", line 51, in
from core.config import Config
File "/home/seba/Documents/UniHCP-inference/core/config.py", line 10, in
from .utils import printlog
File "/home/seba/Documents/UniHCP-inference/core/utils.py", line 19, in
import core.fp16 as fp16
File "/home/seba/Documents/UniHCP-inference/core/fp16/init.py", line 1, in
from .opt import *
File "/home/seba/Documents/UniHCP-inference/core/fp16/opt.py", line 8, in
from .scaler import LossScaler
File "/home/seba/Documents/UniHCP-inference/core/fp16/scaler.py", line 2, in
from torch import inf
ImportError: cannot import name 'inf'

Thanks for the help!