Problem for running the Trained Model

[SHZ2006]

Hi, when I want to run this line of code
from latentfusion.recon.inference import LatentFusionModel

I am getting this error:
cannot import name 'autocast' from 'torch.cuda.amp'

This installation also doesn't help
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch-nightly

It seems the nightly version is not compatible with other toolboxes that you installed in the environment by environment.yml

[keunhong]

Hi,

I don't think you need the nightly version anymore since PyTorch stable has it included.

[SHZ2006]

But still I am getting this error
cannot import name 'autocast' from 'torch.cuda.amp'