- 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', '
|
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
|