From c81c052f4e23c1444c717c36e0dcf1c2dd812fda Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Tue, 2 Dec 2025 04:41:16 -0800 Subject: [PATCH] Add a missing import --- modules/ui_image_generation.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/modules/ui_image_generation.py b/modules/ui_image_generation.py index fa3d6791..b420bc52 100644 --- a/modules/ui_image_generation.py +++ b/modules/ui_image_generation.py @@ -11,7 +11,11 @@ from PIL import Image from PIL.PngImagePlugin import PngInfo from modules import shared, ui, utils -from modules.image_models import load_image_model, unload_image_model +from modules.image_models import ( + get_pipeline_type, + load_image_model, + unload_image_model +) from modules.logging_colors import logger from modules.utils import gradio @@ -627,7 +631,6 @@ def generate(state): Automatically adjusts parameters based on pipeline type. """ import torch - import numpy as np model_name = state['image_model_menu']