update

model_factory.py

@@ -1,6 +1,8 @@
 import torch
 from models.ram import RAM
 
+from huggingface_hub import hf_hub_download
+
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 def get_model():

models/ram.py

@@ -7,8 +7,6 @@ from deepinv.physics import Physics, LinearPhysics, Downsampling
 from deepinv.utils import TensorList
 from deepinv.utils.tensorlist import TensorList
 
-from huggingface_hub import hf_hub_download
-
 cuda = True if torch.cuda.is_available() else False
 Tensor = torch.cuda.FloatTensor if cuda else torch.FloatTensor