BugFix: error on width, height non divisable by 8 and filename typpo

pull/34/head
vahid K. nejad 2023-04-18 13:26:22 +04:00
parent 744347b070
commit 2dc9497d7e
1 changed files with 14 additions and 7 deletions

View File

@ -45,7 +45,12 @@ available_samplers = [
default_prompt = "A psychedelic jungle with trees that have glowing, fractal-like patterns, Simon stalenhag poster 1920s style, street level view, hyper futuristic, 8k resolution, hyper realistic" default_prompt = "A psychedelic jungle with trees that have glowing, fractal-like patterns, Simon stalenhag poster 1920s style, street level view, hyper futuristic, 8k resolution, hyper realistic"
default_negative_prompt = "frames, borderline, text, character, duplicate, error, out of frame, watermark, low quality, ugly, deformed, blur" default_negative_prompt = "frames, borderline, text, character, duplicate, error, out of frame, watermark, low quality, ugly, deformed, blur"
def closest_upper_divisible_by_eight(num):
if num % 8 == 0:
return num
else:
return math.ceil(num/8)*8
def renderTxt2Img(prompt, negative_prompt, sampler, steps, cfg_scale, width, height): def renderTxt2Img(prompt, negative_prompt, sampler, steps, cfg_scale, width, height):
processed = None processed = None
p = StableDiffusionProcessingTxt2Img( p = StableDiffusionProcessingTxt2Img(
@ -204,8 +209,8 @@ def create_zoom_single(
pass pass
assert len(prompts_array) > 0, "prompts is empty" assert len(prompts_array) > 0, "prompts is empty"
width = outputsizeW width = closest_upper_divisible_by_eight(outputsizeW)
height = outputsizeH height = closest_upper_divisible_by_eight(outputsizeH)
current_image = Image.new(mode="RGBA", size=(width, height)) current_image = Image.new(mode="RGBA", size=(width, height))
mask_image = np.array(current_image)[:, :, 3] mask_image = np.array(current_image)[:, :, 3]
@ -376,10 +381,12 @@ def on_ui_tabs():
with gr.Blocks(analytics_enabled=False) as infinite_zoom_interface: with gr.Blocks(analytics_enabled=False) as infinite_zoom_interface:
gr.HTML( gr.HTML(
""" """
<p style='text-align: center'> <p style="text-align: center;">
Text to Video - Infinite zoom effect <a target="_blank" href="https://github.com/v8hid/infinite-zoom-automatic1111-webui"><img src="https://img.shields.io/static/v1?label=github&message=repository&color=blue&style=flat&logo=github&logoColor=white" style="display: inline;" alt="GitHub Repo"/></a>
</p> <a href="https://discord.gg/v2nHqSrWdW"><img src="https://img.shields.io/discord/1095469311830806630?color=blue&label=discord&logo=discord&logoColor=white" style="display: inline;" alt="Discord server"></a>
""" </p>
"""
) )
generate_btn = gr.Button(value="Generate video", variant="primary") generate_btn = gr.Button(value="Generate video", variant="primary")
interrupt = gr.Button(value="Interrupt", elem_id="interrupt_training") interrupt = gr.Button(value="Interrupt", elem_id="interrupt_training")