cpu;

src/main/inference.py

@@ -25,7 +25,7 @@ use_cuda = torch.cuda.is_available()
 
 device =  "cuda" if use_cuda else "cpu"
 
-print("Device",device)
+
 
 class Inference():
     
@@ -101,7 +101,7 @@ class Inference():
         self.generator = load_checkpoint(model=self.generator,
                         path= self.generator_ckpt,
                         optimizer=None,
-                        use_cuda=True,
+                        use_cuda=False,
                         reset_optimizer=True,
                         pretrain=True)
 

utils/wav2lip.py

@@ -33,7 +33,7 @@ def save_checkpoint(model, optimizer,  checkpoint_dir,epoch, savename):
 
 
 def _load(checkpoint_path, use_cuda):
-    print("use : cuda", use_cuda)
+ 
     if use_cuda:
         checkpoint = torch.load(checkpoint_path)
     else: