mart9992 commited on
Commit
b8303b0
1 Parent(s): 84c93fa
Files changed (1) hide show
  1. handler.py +3 -2
handler.py CHANGED
@@ -2,9 +2,10 @@ import string
2
  import warnings
3
  warnings.filterwarnings('ignore')
4
  import subprocess, io, os, sys, time
 
5
 
6
- os.environ["XFORMERS_DISABLE_FLASH_ATTN"] = "1"
7
- result = subprocess.run(['pip', 'install', 'git+https://github.com/facebookresearch/xformers.git@main#egg=xformers'], check=True)
8
 
9
  from entklei import get_nude
10
  from scipy.ndimage import binary_dilation
 
2
  import warnings
3
  warnings.filterwarnings('ignore')
4
  import subprocess, io, os, sys, time
5
+ from dw_pose.main import dwpose
6
 
7
+ # os.environ["XFORMERS_DISABLE_FLASH_ATTN"] = "1"
8
+ result = subprocess.run(['pip', 'install', 'xformers'], check=True)
9
 
10
  from entklei import get_nude
11
  from scipy.ndimage import binary_dilation