LivePortrait2

Running on Zero

yerang commited on Oct 16

Commit

379f058

•

1 Parent(s): 69f7f0d

Update stf_utils.py

Files changed (1) hide show

stf_utils.py CHANGED Viewed

@@ -77,7 +77,7 @@ class STFPipeline:
         checkpoint_path: str = "089.pth",
         root_path: str = "works",
         wavlm_path: str = "microsoft/wavlm-large",
-        device: str = "cuda:0"
     ):
         self.device = device
         self.stf_path = stf_path

         checkpoint_path: str = "089.pth",
         root_path: str = "works",
         wavlm_path: str = "microsoft/wavlm-large",
+        device: str = "cuda"
     ):
         self.device = device
         self.stf_path = stf_path