From ba7d0cbb34ffd4f13176ac4c8e59a864673f7461 Mon Sep 17 00:00:00 2001 From: Extraltodeus Date: Thu, 23 May 2024 07:00:43 +0200 Subject: [PATCH] Add files via upload --- experimental_temperature.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/experimental_temperature.py b/experimental_temperature.py index 3e0237e..d1ecd31 100644 --- a/experimental_temperature.py +++ b/experimental_temperature.py @@ -95,7 +95,7 @@ class ExperimentalTemperaturePatch: RETURN_NAMES = ("Model","String",) FUNCTION = "patch" - CATEGORY = "model_patches/Automatic_CFG" + CATEGORY = "model_patches/Standalone_temperature_patches/Automatic_CFG" def patch(self, model, Temperature, Attention, **kwargs): m = model.clone() @@ -129,7 +129,7 @@ class CLIPTemperaturePatch: RETURN_TYPES = ("CLIP",) FUNCTION = "merge" - CATEGORY = "model_patches/Automatic_CFG" + CATEGORY = "model_patches/Standalone_temperature_patches/Automatic_CFG" def patch(self, clip, Temperature): def custom_optimized_attention(device, mask=None, small_input=True): @@ -153,7 +153,7 @@ class CLIPTemperaturePatch: clip_encoder_instance = m.cond_stage_model.clip_l.transformer.text_model.encoder clip_encoder_instance.forward = types.MethodType(new_forward, clip_encoder_instance) - + if getattr(m.cond_stage_model, f"clip_g", None) is not None: clip_encoder_instance_g = m.cond_stage_model.clip_g.transformer.text_model.encoder clip_encoder_instance_g.forward = types.MethodType(new_forward, clip_encoder_instance_g)