Skip to content

Commit ae4644c

Browse files
committed
Bump up image quality default
1 parent 7c502e5 commit ae4644c

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

predict.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -72,9 +72,7 @@ def setup(self) -> None: # pyright: ignore
7272
download_base_weights(SAFETY_URL, SAFETY_CACHE_PATH)
7373
self.safety_checker = StableDiffusionSafetyChecker.from_pretrained(
7474
SAFETY_CACHE_PATH, torch_dtype=torch.float16
75-
).to(
76-
"cuda"
77-
) # pyright: ignore
75+
).to("cuda") # pyright: ignore
7876
self.feature_extractor = cast(
7977
CLIPImageProcessor, CLIPImageProcessor.from_pretrained(FEATURE_EXTRACTOR)
8078
)
@@ -242,7 +240,7 @@ def predict( # pyright: ignore
242240
),
243241
output_quality: int = Input(
244242
description="Quality when saving the output images, from 0 to 100. 100 is best quality, 0 is lowest quality. Not relevant for .png outputs",
245-
default=80,
243+
default=90,
246244
ge=0,
247245
le=100,
248246
),

0 commit comments

Comments
 (0)