remove commented out line.

entrpn · entrpn · commit 618550c1835a · 2025-03-13T20:51:15.000Z
diff --git a/src/maxdiffusion/models/attention_flax.py b/src/maxdiffusion/models/attention_flax.py
@@ -206,7 +206,7 @@ def cudnn_flash_attention(
     def wrap_flash_attention(query, key, value):
       return jax.vmap(self.dpa_layer)(query, key, value, mask=None)
 
-    out = wrap_flash_attention(query, key, value)#self.dpa_layer(query, key, value, mask=None)
+    out = wrap_flash_attention(query, key, value)
     return self.reshape_data_from_cudnn_flash(out)
 
   def apply_attention_dot(self, query: Array, key: Array, value: Array):
