Upload folder using huggingface_hub

sampler.py

@@ -93,7 +93,10 @@ class BaseSampler:
         log_str = f'Building the diffusion model with length: {self.configs.diffusion.params.steps}...'
         self.write_log(log_str)
         self.base_diffusion = util_common.instantiate_from_config(self.configs.diffusion)
-        model = util_common.instantiate_from_config(self.configs.model).cuda()
+        model = util_common.instantiate_from_config(self.configs.model)
+        # gpu test
+        if torch.cuda.is_available():
+            model = model.cuda()
         ckpt_path =self.configs.model.ckpt_path
         assert ckpt_path is not None
         self.write_log(f'Loading Diffusion model from {ckpt_path}...')