FoleyCrafter

Running on Zero

unknown commited on Jun 28

Commit

14c3956

•

1 Parent(s): ab7f7d5

cuda

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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.pipeline = None

         self.savedir_sample = os.path.join(self.savedir, "sample")
         os.makedirs(self.savedir, exist_ok=True)
+        self.device = "cuda"
         self.pipeline = None