Commit 4738486d authored by brkirch's avatar brkirch

Support for hypernetworks with --upcast-sampling

parent ea9bd9fc
...@@ -380,8 +380,8 @@ def apply_single_hypernetwork(hypernetwork, context_k, context_v, layer=None): ...@@ -380,8 +380,8 @@ def apply_single_hypernetwork(hypernetwork, context_k, context_v, layer=None):
layer.hyper_k = hypernetwork_layers[0] layer.hyper_k = hypernetwork_layers[0]
layer.hyper_v = hypernetwork_layers[1] layer.hyper_v = hypernetwork_layers[1]
context_k = hypernetwork_layers[0](context_k) context_k = devices.cond_cast_unet(hypernetwork_layers[0](devices.cond_cast_float(context_k)))
context_v = hypernetwork_layers[1](context_v) context_v = devices.cond_cast_unet(hypernetwork_layers[1](devices.cond_cast_float(context_v)))
return context_k, context_v return context_k, context_v
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment