Skip to content

Commit aa247e6

Browse files
committed
use refined model by default
1 parent 895c47f commit aa247e6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ldm/generate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -896,7 +896,7 @@ def apply_textmask(self, image_path:str, prompt:str, callback, threshold:float=0
896896
assert os.path.exists(image_path), '** "{image_path}" not found. Please enter the name of an existing image file to mask **'
897897
basename,_ = os.path.splitext(os.path.basename(image_path))
898898
if self.txt2mask is None:
899-
self.txt2mask = Txt2Mask(device = self.device)
899+
self.txt2mask = Txt2Mask(device = self.device, refined=True)
900900
segmented = self.txt2mask.segment(image_path,prompt)
901901
trans = segmented.to_transparent()
902902
inverse = segmented.to_transparent(invert=True)

0 commit comments

Comments
 (0)