Function make_images_internal

ui/easydiffusion/tasks/render_images.py:166–216

(
    context,
    req: GenerateImageRequest,
    task_data: RenderTaskData,
    models_data: ModelsData,
    output_format: OutputFormatData,
    save_data: SaveToDiskData,
    data_queue: queue.Queue,
    task_temp_images: list,
    step_callback,
    task,
)

def make_images_internal(
    context,
    req: GenerateImageRequest,
    task_data: RenderTaskData,
    models_data: ModelsData,
    output_format: OutputFormatData,
    save_data: SaveToDiskData,
    data_queue: queue.Queue,
    task_temp_images: list,
    step_callback,
    task,
):
    from easydiffusion.backend_manager import backend

    # prep the nsfw_filter
    if task_data.block_nsfw:
        filter_nsfw([Image.new("RGB", (1, 1))])  # hack - ensures that the model is available

    images = generate_images_internal(
        context,
        req,
        task_data,
        models_data,
        output_format,
        data_queue,
        task_temp_images,
        step_callback,
        task_data.stream_image_progress,
        task_data.stream_image_progress_interval,
    )
    user_stopped = isinstance(task.error, StopAsyncIteration)

    filters, filter_params = task_data.filters, task_data.filter_params
    if len(filters) > 0 and not user_stopped:
        filtered_images = backend.filter_images(context, images, filters, filter_params, input_type="base64")
    else:
        filtered_images = images

    if task_data.block_nsfw:
        filtered_images = filter_nsfw(filtered_images)

    if save_data.save_to_disk_path is not None:
        images_pil = [base64_str_to_img(img) for img in images]
        filtered_images_pil = [base64_str_to_img(img) for img in filtered_images]
        save_images_to_disk(images_pil, filtered_images_pil, req, task_data, models_data, output_format, save_data)

    seeds = [*range(req.seed, req.seed + len(images))]
    if task_data.show_only_filtered_image or filtered_images is images:
        return filtered_images, seeds
    else:
        return images + filtered_images, seeds + seeds


def generate_images_internal(