cuda

app.py

@@ -64,7 +64,7 @@ class FoleyController:
         self.savedir_sample = os.path.join(self.savedir, "sample")
         os.makedirs(self.savedir, exist_ok=True)
 
-        self.device = "cuda" if torch.cuda.is_available() else "cpu"
+        self.device = "cuda"
 
         self.pipeline = None
 
@@ -120,6 +120,12 @@ class FoleyController:
 
         self.pipeline.load_ip_adapter(fc_ckpt, subfolder='semantic', weight_name='semantic_adapter.bin', image_encoder_folder=None)
 
+        # move to gpu
+        self.time_detector = self.time_detector.to(self.device)
+        self.pipeline = self.pipeline.to(self.device)
+        self.vocoder = self.vocoder.to(self.device)
+        self.image_encoder = self.image_encoder.to(self.device)
+
         gr.Info("Load Finish!")
         print("Load Finish!")
         self.loaded = True
@@ -138,12 +144,7 @@ class FoleyController:
         sample_step_slider,
         cfg_scale_slider,
         seed_textbox, 
-    ):
-        # move to gpu
-        self.time_detector = self.time_detector.to(self.device)
-        self.pipeline = self.pipeline.to(self.device)
-        self.vocoder = self.vocoder.to(self.device)
-        self.image_encoder = self.image_encoder.to(self.device)
+    ): 
 
         vision_transform_list = [
             torchvision.transforms.Resize((128, 128)),