From aa247e68be48c443d322be5e0d1f5261042a6a20 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Wed, 2 Nov 2022 18:29:34 -0400 Subject: [PATCH] use refined model by default --- ldm/generate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ldm/generate.py b/ldm/generate.py index 4733551261..3190229c3e 100644 --- a/ldm/generate.py +++ b/ldm/generate.py @@ -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 **' basename,_ = os.path.splitext(os.path.basename(image_path)) 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) trans = segmented.to_transparent() inverse = segmented.to_transparent(invert=True)