1580158
/
ebsynth_utility
mirror of https://github.com/s9roll7/ebsynth_utility.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

When the path of the python virtual environment is not venv, transparent_background will not be found here. This modification finds the virtual environment path as the base path from the environment variable VIRTUAL_ENV

pull/67/head
allenhooo 2023-05-30 20:18:06 +08:00
parent 2f28fcde5d
commit d688bc4e63
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
stage1.py
View File

@ -147,7 +147,8 @@ def create_mask_clipseg(input_dir, output_dir, clipseg_mask_prompt, clipseg_excl
def create_mask_transparent_background(input_dir, output_dir, tb_use_fast_mode, tb_use_jit, st1_mask_threshold): def create_mask_transparent_background(input_dir, output_dir, tb_use_fast_mode, tb_use_jit, st1_mask_threshold):
fast_str = " --fast" if tb_use_fast_mode else "" fast_str = " --fast" if tb_use_fast_mode else ""
jit_str = " --jit" if tb_use_jit else "" jit_str = " --jit" if tb_use_jit else ""
bin_path = os.path.join("venv", "Scripts") venv = os.environ['VIRTUAL_ENV']
bin_path = os.path.join(venv, "Scripts")
bin_path = os.path.join(bin_path, "transparent-background") bin_path = os.path.join(bin_path, "transparent-background")
if os.path.isfile(bin_path) or os.path.isfile(bin_path + ".exe"): if os.path.isfile(bin_path) or os.path.isfile(bin_path + ".exe"):