Fix (graph): Changed SDPA import

Xilinx · Nov 20, 2024 · fc2e965 · fc2e965
1 parent 7b20742
commit fc2e965
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas/graph/standardize.py b/src/brevitas/graph/standardize.py
@@ -10,7 +10,7 @@
 from brevitas.fx import GraphModule
 from brevitas.fx import immutable_dict
 from brevitas.fx import Node
-from brevitas.nn.quant_sdpa import ScaledDotProductAttention
+from brevitas.nn import ScaledDotProductAttention
 
 from .base import FnToModule
 from .base import GraphTransform