Support for hypernetworks with --upcast-sampling (4738486d) · Commits · github_fork / Stable Diffusion Webui

modules/hypernetworks/hypernetwork.py

+2 −2

Original line number	Diff line number	Diff line
		@@ -380,8 +380,8 @@ def apply_single_hypernetwork(hypernetwork, context_k, context_v, layer=None):
		layer.hyper_k = hypernetwork_layers[0]
		layer.hyper_v = hypernetwork_layers[1]

		context_k = hypernetwork_layers[0](context_k)
		context_v = hypernetwork_layers[1](context_v)
		context_k = devices.cond_cast_unet(hypernetwork_layers[0](devices.cond_cast_float(context_k)))
		context_v = devices.cond_cast_unet(hypernetwork_layers[1](devices.cond_cast_float(context_v)))
		return context_k, context_v