Update model/FGResQ.py

model/FGResQ.py

@@ -83,7 +83,7 @@ class DualBranch(nn.Module):
         Load pre-trained weights, including the CLIP model and classification head.
         """
         # Load state dictionary
-        state_dict = torch.load(state_dict_path)
+        state_dict = torch.load(state_dict_path, map_location=torch.device('cpu'))
         
         # Separate weights for CLIP model and classification head
         clip_state_dict = {}