diff --git a/example/generate_controlnet_masks.py b/example/generate_controlnet_masks.py index 4994ed9..08aeefa 100644 --- a/example/generate_controlnet_masks.py +++ b/example/generate_controlnet_masks.py @@ -25,14 +25,12 @@ async def main(): async with API() as api_handler: api = api_handler.api - with open(d / "image.png", "rb") as f: - image = base64.b64encode(f.read()).decode() + image = base64.b64encode((d / "image.png").read_bytes()).decode() for controlnet in ControlNetModel: try: _, img = await api.low_level.generate_controlnet_mask(controlnet, image) - with open(d / f"image_{controlnet.value}.png", "wb") as f: - f.write(img) + (d / f"image_{controlnet.value}.png").write_bytes(img) print(f"Generated with {controlnet.value}") time.sleep(2) diff --git a/example/generate_image.py b/example/generate_image.py index 0094799..218d52f 100644 --- a/example/generate_image.py +++ b/example/generate_image.py @@ -22,13 +22,13 @@ async def main(): api = api_handler.api preset = ImagePreset() + preset.seed = 42 # multiple images # preset.n_samples = 4 i = 0 async for _, img in api.high_level.generate_image("1girl", ImageModel.Anime_Full, preset): - with open(d / f"image_1_{i}.png", "wb") as f: - f.write(img) + (d / f"image_1_{i}.png").write_bytes(img) i += 1 @@ -37,8 +37,7 @@ async def main(): preset.resolution = (128, 256) async for _, img in api.high_level.generate_image("1girl", ImageModel.Anime_Full, preset): - with open(d / "image_2.png", "wb") as f: - f.write(img) + (d / "image_2.png").write_bytes(img) # furry model preset.resolution = ImageResolution.Normal_Square @@ -46,8 +45,7 @@ async def main(): preset.uc_preset = UCPreset.Preset_Low_Quality async for _, img in api.high_level.generate_image("female, species:human", ImageModel.Furry, preset): - with open(d / "image_3.png", "wb") as f: - f.write(img) + (d / "image_3.png").write_bytes(img) if __name__ == "__main__": diff --git a/example/generate_image_test_samplers.py b/example/generate_image_test_samplers.py index cf673b7..1f6a097 100644 --- a/example/generate_image_test_samplers.py +++ b/example/generate_image_test_samplers.py @@ -24,14 +24,14 @@ async def main(): api = api_handler.api preset = ImagePreset() + preset.seed = 42 for sampler in ImageSampler: preset.sampler = sampler try: async for _, img in api.high_level.generate_image("1girl", ImageModel.Anime_Full, preset): - with open(d / f"image_{sampler.value}.png", "wb") as f: - f.write(img) + (d / f"image_{sampler.value}").write_bytes(img) print(f"Generated with {sampler.value}") time.sleep(2) diff --git a/example/generate_image_with_controlnet.py b/example/generate_image_with_controlnet.py index 40e0c69..60c3f57 100644 --- a/example/generate_image_with_controlnet.py +++ b/example/generate_image_with_controlnet.py @@ -23,8 +23,7 @@ async def main(): async with API() as api_handler: api = api_handler.api - with open(d / "image.png", "rb") as f: - image = base64.b64encode(f.read()).decode() + image = base64.b64encode((d / "image.png").read_bytes()).decode() controlnet = ControlNetModel.Form_Lock _, mask = await api.low_level.generate_controlnet_mask(controlnet, image) @@ -33,11 +32,11 @@ async def main(): preset.controlnet_model = controlnet preset.controlnet_condition = base64.b64encode(mask).decode() preset.controlnet_strength = 1.5 + preset.seed = 42 # NOTE: for some reasons, the images with controlnet are slightly different async for _, img in api.high_level.generate_image("1girl", ImageModel.Anime_Full, preset): - with open(d / "image_with_controlnet.png", "wb") as f: - f.write(img) + (d / "image_with_controlnet.png").write_bytes(img) if __name__ == "__main__": diff --git a/example/generate_image_with_img2img.py b/example/generate_image_with_img2img.py index 976d9b5..3fd4d61 100644 --- a/example/generate_image_with_img2img.py +++ b/example/generate_image_with_img2img.py @@ -22,20 +22,19 @@ async def main(): async with API() as api_handler: api = api_handler.api - with open(d / "image.png", "rb") as f: - image = base64.b64encode(f.read()).decode() + image = base64.b64encode((d / "image.png").read_bytes()).decode() preset = ImagePreset() preset.noise = 0.1 # note that steps = 28, not 50, which mean strength needs to be adjusted accordingly preset.strength = 0.5 preset.image = image + preset.seed = 42 async for _, img in api.high_level.generate_image( "1girl", ImageModel.Anime_Full, preset, ImageGenerationType.IMG2IMG ): - with open(d / "image_with_img2img.png", "wb") as f: - f.write(img) + (d / "image_with_img2img.png").write_bytes(img) if __name__ == "__main__": diff --git a/example/generate_image_with_inpainting.py b/example/generate_image_with_inpainting.py index b876009..46274a7 100644 --- a/example/generate_image_with_inpainting.py +++ b/example/generate_image_with_inpainting.py @@ -26,12 +26,8 @@ async def main(): mask = base64.b64encode((d / "inpainting_mask.png").read_bytes()).decode() preset = ImagePreset() - preset.noise = 0.1 - # note that steps = 28, not 50, which mean strength needs to be adjusted accordingly - preset.strength = 0.5 preset.image = image preset.mask = mask - preset.add_original_image = False preset.seed = 42 async for _, img in api.high_level.generate_image( diff --git a/example/upscale_image.py b/example/upscale_image.py index 0468d9e..978dbc1 100644 --- a/example/upscale_image.py +++ b/example/upscale_image.py @@ -26,15 +26,13 @@ async def main(): async with API() as api_handler: api = api_handler.api - with open(d / "image.png", "rb") as f: - image = base64.b64encode(f.read()).decode() + image = base64.b64encode((d / "image.png").read_bytes()).decode() # disable the type check on scale in _low_level.py to check on float values for scale in (2, 4): try: _, img = await api.low_level.upscale_image(image, *image_size, scale) - with open(f"image_upscaled_{scale}.png", "wb") as f: - f.write(img) + (d / f"image_upscaled_{scale}.png").write_bytes(img) print(f"Generated upscale {scale}") time.sleep(2)