Upload folder using huggingface_hub

sampler.py

@@ -117,7 +117,7 @@ class BaseSampler:
             self.autoencoder = None
 
     def load_model(self, model, ckpt_path=None):
-        state = torch.load(ckpt_path, map_location=f"cuda:{self.rank}")
+        state = torch.load(ckpt_path, map_location=torch.device('cpu'))
         if 'state_dict' in state:
             state = state['state_dict']
         util_net.reload_model(model, state)