スポンサーリンク
I have used diffusers and LoRA to generate various images.
technoxs-stacker.hatenablog.com
This time I tried to see if I could create a high resolution image
This article is just a reminder, but I hope it will be useful for everyone!
contents
スポンサーリンク
abstract
How to use diffusers and upscaler on Google colab to create high resolution images.
1.requirement
Google Colab
Diffusers
transformers==4.26.0
model : chilloutmix_NiPrunedFp32Fix
upscaler:stabilityai/stable-diffusion-x4-upscaler
2.code
from diffusers import StableDiffusionUpscalePipeline def upscale(pipe, prompt, img, step=100): return pipe(prompt=prompt, image=img, num_inference_steps=step, guidance_scale=0).images[0] # make pipeline pipe_upscaler = StableDiffusionUpscalePipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler", torch_dtype=torch.float16, revision="fp16") pipe_upscaler = pipe_upscaler.to("cuda") prompt = "prompt text" im = Image.open(save_path).convert("RGB")#save_path:path to image low_res_img = im.resize((256, 256)) step = 200 pipe_upscaler(prompt=prompt, image=low_res_img, num_inference_steps=step, guidance_scale=0).images[0]
3.result
normal(552 x 552 pixel)
high resolution(1024 x 1024 pixel)
The image is clearer after the high resolution when compared to the enlarged image.
left:normal / right:high resolution
スポンサーリンク