Hugging Face's logo

Spaces:
yerang
/
LivePortrait2
Running on Zero

App Files Community
yerang commited on
Commit
e6acc0a
·
verified ·
1 Parent(s): eef1f4f

Update stf/stf-api-alternative/src/stf_alternative/dataset.py

Browse files

Add 'cv2.setNumThreads(0); cv2.ocl.setUseOpenCL(False)'

Files changed (1) hide show
  1. stf/stf-api-alternative/src/stf_alternative/dataset.py +1 -1
stf/stf-api-alternative/src/stf_alternative/dataset.py CHANGED
@@ -3,7 +3,7 @@ import random
3
  from glob import glob
4
  from pathlib import Path
5
 
6
- import cv2
7
  import numpy as np
8
  import pandas as pd
9
  from torch.utils.data import Dataset
 
3
  from glob import glob
4
  from pathlib import Path
5
 
6
+ import cv2,cv2.setNumThreads(0); cv2.ocl.setUseOpenCL(False)
7
  import numpy as np
8
  import pandas as pd
9
  from torch.utils.data import Dataset