...

gradio_demo/barc_demo_v3.py

@@ -168,7 +168,7 @@ def run_barc_inference(input_image, bbox=None):
     print(path_model_file_complete)
     assert os.path.isfile(path_model_file_complete)
     print('Loading model weights from file: {}'.format(path_model_file_complete))
-    checkpoint_complete = torch.load(path_model_file_complete)
+    checkpoint_complete = torch.load(path_model_file_complete, map_location=device)
     state_dict_complete = checkpoint_complete['state_dict']
     complete_model.load_state_dict(state_dict_complete, strict=False)        
     complete_model = complete_model.to(device)

src/priors/normalizing_flow_prior/normalizing_flow_prior.py

@@ -48,7 +48,8 @@ class NormalizingFlowPrior(nn.Module):
         if weight_path is None:
             weight_path = self.param_dict['path_pretrained']
         print(' normalizing flow pose prior: loading {}..'.format(weight_path))
-        pretrained_dict = torch.load(weight_path)['model_state_dict']
+        device = get_cfg_global_updated().device
+        pretrained_dict = torch.load(weight_path, map_location=device)['model_state_dict']
         self.model_inn.load_state_dict(pretrained_dict, strict=True)
 
     def subnet_fc(self, c_in, c_out):