Update nodes.py
This commit is contained in:
@@ -2,7 +2,7 @@ import torch
|
||||
import math
|
||||
import types
|
||||
|
||||
EPSILON = 1e-4
|
||||
EPSILON = 1e-16
|
||||
|
||||
SD_layer_dims = {
|
||||
"SD1" : {"input_1": 4096,"input_2": 4096,"input_4": 1024,"input_5": 1024,"input_7": 256,"input_8": 256,"middle_0": 64,"output_3": 256,"output_4": 256,"output_5": 256,"output_6": 1024,"output_7": 1024,"output_8": 1024,"output_9": 4096,"output_10": 4096,"output_11": 4096},
|
||||
@@ -11,6 +11,7 @@ SD_layer_dims = {
|
||||
}
|
||||
|
||||
def should_scale(mname,lname,q2):
|
||||
if mname == "": return False
|
||||
if mname != "Disabled" and lname in SD_layer_dims[mname]:
|
||||
return q2 != SD_layer_dims[mname][lname]
|
||||
return False
|
||||
@@ -49,7 +50,7 @@ class UnetTemperaturePatch:
|
||||
required_inputs = {}
|
||||
required_inputs["model"] = ("MODEL",)
|
||||
required_inputs["Temperature"] = ("FLOAT", {"default": 1.0, "min": 0.0, "max": 10.0, "step": 0.01, "round": 0.01})
|
||||
required_inputs["Attention"] = (["self","cross","both"],)
|
||||
required_inputs["Attention"] = (["both","self","cross"],)
|
||||
required_inputs["Dynamic_Scale_Attention"] = (["Disabled","SDXL","SD1"],)
|
||||
return {"required": required_inputs}
|
||||
|
||||
@@ -115,4 +116,4 @@ class CLIPTemperaturePatch:
|
||||
clip_encoder_instance_g = clip.cond_stage_model.clip_g.transformer.text_model.encoder
|
||||
clip_encoder_instance_g.forward = types.MethodType(new_forward, clip_encoder_instance_g)
|
||||
|
||||
return (clip,)
|
||||
return (clip,)
|
||||
|
||||
Reference in New Issue
Block a user