Update src/diffusers/models/attention_processor.py

a-r-r-o-w · sayakpaul · yiyixuxu · commit e96d5ad599e6 · 2024-10-21T20:51:13.000+02:00
Co-authored-by: Sayak Paul &lt;spsayakpaul@gmail.com&gt;
diff --git a/src/diffusers/models/attention_processor.py b/src/diffusers/models/attention_processor.py
@@ -252,7 +252,7 @@ def __init__(
                 self.norm_added_k = RMSNorm(dim_head, eps=eps)
             else:
                 raise ValueError(
-                    f"unknown qk_norm: {qk_norm}. Should be None,'layer_norm','fp32_layer_norm','rms_norm'"
+                    f"unknown qk_norm: {qk_norm}. Should be one of `None,'layer_norm','fp32_layer_norm','rms_norm'`"
                 )
         else:
             self.norm_added_q = None

Original file line number	Diff line number	Diff line change
`@@ -252,7 +252,7 @@ def __init__(`
`252`	`252`	`self.norm_added_k = RMSNorm(dim_head, eps=eps)`
`253`	`253`	`else:`
`254`	`254`	`raise ValueError(`
`255`		`- f"unknown qk_norm: {qk_norm}. Should be None,'layer_norm','fp32_layer_norm','rms_norm'"`
	`255`	+ f"unknown qk_norm: {qk_norm}. Should be one of `None,'layer_norm','fp32_layer_norm','rms_norm'`"
`256`	`256`	`)`
`257`	`257`	`else:`
`258`	`258`	`self.norm_added_q = None`