diff --git a/Gen_3D_Modules/Stable3DGen/trellis/backend_config.py b/Gen_3D_Modules/Stable3DGen/trellis/backend_config.py index bd72c38..70fcccf 100644 --- a/Gen_3D_Modules/Stable3DGen/trellis/backend_config.py +++ b/Gen_3D_Modules/Stable3DGen/trellis/backend_config.py @@ -48,7 +48,7 @@ def _try_import_sageattention() -> bool: try: import torch.nn.functional as F from sageattention import sageattn - F.scaled_dot_product_attention = sageattn + #F.scaled_dot_product_attention = sageattn #import sageattention return True except ImportError: