From fc2e9655fa08fa89ebeb8252220389d5c3313e8f Mon Sep 17 00:00:00 2001 From: Nick Fraser Date: Wed, 20 Nov 2024 17:53:25 +0000 Subject: [PATCH] Fix (graph): Changed SDPA import --- src/brevitas/graph/standardize.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/brevitas/graph/standardize.py b/src/brevitas/graph/standardize.py index 4c9233e08..7d15b12ed 100644 --- 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