Replace realesrgan to the Stable Diffusion latent upscaler. Remove unnecessary libraries.

This commit is contained in:
hodanov
2024-04-21 21:34:57 +09:00
parent b1de1d4cb5
commit d531c25bd0
11 changed files with 71 additions and 244 deletions
+2 -4
View File
@@ -15,8 +15,7 @@ def main(
batch_size: int = 1,
steps: int = 20,
seed: int = -1,
upscaler: str = "",
use_face_enhancer: str = "False",
use_upscaler: str = "False",
fix_by_controlnet_tile: str = "False",
output_format: str = "png",
base_image_url: str = "",
@@ -38,8 +37,7 @@ def main(
batch_size=batch_size,
steps=steps,
seed=seed_generated,
upscaler=upscaler,
use_face_enhancer=use_face_enhancer == "True",
use_upscaler=use_upscaler == "True",
fix_by_controlnet_tile=fix_by_controlnet_tile == "True",
output_format=output_format,
base_image_url=base_image_url,
+2 -4
View File
@@ -17,8 +17,7 @@ def main(
batch_size: int = 1,
steps: int = 20,
seed: int = -1,
upscaler: str = "",
use_face_enhancer: str = "False",
use_upscaler: str = "",
fix_by_controlnet_tile: str = "False",
output_format: str = "png",
):
@@ -41,8 +40,7 @@ def main(
batch_size=batch_size,
steps=steps,
seed=seed_generated,
upscaler=upscaler,
use_face_enhancer=use_face_enhancer == "True",
use_upscaler=use_upscaler == "True",
fix_by_controlnet_tile=fix_by_controlnet_tile == "True",
output_format=output_format,
)
+2 -4
View File
@@ -16,8 +16,7 @@ def main(
samples: int = 5,
steps: int = 20,
seed: int = -1,
upscaler: str = "",
use_face_enhancer: str = "False",
use_upscaler: str = "False",
output_format: str = "png",
):
"""
@@ -38,8 +37,7 @@ def main(
width=width,
steps=steps,
seed=seed_generated,
upscaler=upscaler,
use_face_enhancer=use_face_enhancer == "True",
use_upscaler=use_upscaler == "True",
output_format=output_format,
)
util.save_images(directory, images, seed_generated, i, output_format)