update

models/ram.py

@@ -10,7 +10,7 @@ from deepinv.utils.tensorlist import TensorList
 from huggingface_hub import hf_hub_download
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
-Tensor = torch.cuda.FloatTensor if cuda else torch.FloatTensor
+Tensor = torch.cuda.FloatTensor if torch.cuda.is_available() else torch.FloatTensor
 
 class RAM(nn.Module):
     r"""