Spaces:
Running
on
Zero
Running
on
Zero
Update stf_utils.py
Browse files- stf_utils.py +1 -1
stf_utils.py
CHANGED
|
@@ -77,7 +77,7 @@ class STFPipeline:
|
|
| 77 |
checkpoint_path: str = "089.pth",
|
| 78 |
root_path: str = "works",
|
| 79 |
wavlm_path: str = "microsoft/wavlm-large",
|
| 80 |
-
device: str = "cuda
|
| 81 |
):
|
| 82 |
self.device = device
|
| 83 |
self.stf_path = stf_path
|
|
|
|
| 77 |
checkpoint_path: str = "089.pth",
|
| 78 |
root_path: str = "works",
|
| 79 |
wavlm_path: str = "microsoft/wavlm-large",
|
| 80 |
+
device: str = "cuda"
|
| 81 |
):
|
| 82 |
self.device = device
|
| 83 |
self.stf_path = stf_path
|