diff --git a/ui/media/js/engine.js b/ui/media/js/engine.js index d3c20328..f357b8bc 100644 --- a/ui/media/js/engine.js +++ b/ui/media/js/engine.js @@ -743,6 +743,8 @@ "use_face_correction": 'string', "use_upscale": 'string', "use_vae_model": 'string', + "use_hypernetwork_model": 'string', + "hypernetwork_strength": 'number', } // Higer values will result in... diff --git a/ui/media/js/main.js b/ui/media/js/main.js index de5c988b..7192b0d3 100644 --- a/ui/media/js/main.js +++ b/ui/media/js/main.js @@ -879,6 +879,10 @@ function getCurrentUserRequest() { if (useUpscalingField.checked) { newTask.reqBody.use_upscale = upscaleModelField.value } + if (hypernetworkModelField.value) { + newTask.reqBody.use_hypernetwork_model = hypernetworkModelField.value + newTask.reqBody.hypernetwork_strength = parseFloat(hypernetworkStrengthField.value) + } return newTask } @@ -1185,7 +1189,7 @@ async function getModels() { const stableDiffusionOptions = modelsOptions['stable-diffusion'] const vaeOptions = modelsOptions['vae'] - const hypernetworkOptions = modelOptions['hypernetwork'] + const hypernetworkOptions = modelsOptions['hypernetwork'] vaeOptions.unshift('') // add a None option hypernetworkOptions.unshift('') // add a None option diff --git a/ui/sd_internal/runtime.py b/ui/sd_internal/runtime.py index 2e0f8ce0..733603ac 100644 --- a/ui/sd_internal/runtime.py +++ b/ui/sd_internal/runtime.py @@ -28,7 +28,6 @@ from gfpgan import GFPGANer from basicsr.archs.rrdbnet_arch import RRDBNet from realesrgan import RealESRGANer -from . import hypernetwork from server import HYPERNETWORK_MODEL_EXTENSIONS# , STABLE_DIFFUSION_MODEL_EXTENSIONS, VAE_MODEL_EXTENSIONS from threading import Lock @@ -450,6 +449,11 @@ def is_hypernetwork_reload_necessary(req: Request): return needs_model_reload def load_hypernetwork(): + if thread_data.test_sd2: + # Not yet supported in SD2 + return + + from . import hypernetwork if thread_data.hypernetwork_file is not None: try: loaded = False