Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

use refined model by default

Browse Source
This commit is contained in:
Lincoln Stein 2022-11-02 18:29:34 -04:00
parent 895c47fd11
commit aa247e68be
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ldm/generate.py
View File

@ -896,7 +896,7 @@ class Generate:
assert os.path.exists(image_path), '** "{image_path}" not found. Please enter the name of an existing image file to mask **' assert os.path.exists(image_path), '** "{image_path}" not found. Please enter the name of an existing image file to mask **'
basename,_ = os.path.splitext(os.path.basename(image_path)) basename,_ = os.path.splitext(os.path.basename(image_path))
if self.txt2mask is None: if self.txt2mask is None:
self.txt2mask = Txt2Mask(device = self.device) self.txt2mask = Txt2Mask(device = self.device, refined=True)
segmented = self.txt2mask.segment(image_path,prompt) segmented = self.txt2mask.segment(image_path,prompt)
trans = segmented.to_transparent() trans = segmented.to_transparent()
inverse = segmented.to_transparent(invert=True) inverse = segmented.to_transparent(invert=True)