INNER CODE UNIT · Python
sim
Extraltodeus/ComfyUI-AutomaticCFG · experimental_temperature.py:65
sim = sim.div(self.temperature if self.temperature > 0 else abs_mean(sim)).softmax(dim=-1)
out = einsum('b i j, b j d -> b i d', sim.to(v.dtype), v)
out = (
out.unsqueeze(0)
.reshape(b, heads, -1, dim_head)
.permute(0, 2, 1, 3)
.reshape(b, -1, heads * dim_head)
)
return out
layers_SD15 = {
"input":[1,2,4,5,7,8],
"middle":[0],
"output":[3,4,5,6,7,8,9,10,11],
}
layers_SDXL = {