Update core/graph_mamba.py
Browse files- core/graph_mamba.py +3 -3
core/graph_mamba.py
CHANGED
@@ -350,9 +350,9 @@ def create_astrocyte_config():
|
|
350 |
'input_dim': 1433
|
351 |
}
|
352 |
|
353 |
-
#
|
354 |
-
AstrocyteGraphMamba =
|
355 |
-
GraphMamba =
|
356 |
HybridGraphMamba = LegacyGraphMamba
|
357 |
QuantumEnhancedGraphMamba = LegacyGraphMamba
|
358 |
create_regularized_config = create_astrocyte_config
|
|
|
350 |
'input_dim': 1433
|
351 |
}
|
352 |
|
353 |
+
# Use simple working version for now
|
354 |
+
AstrocyteGraphMamba = LegacyGraphMamba
|
355 |
+
GraphMamba = LegacyGraphMamba
|
356 |
HybridGraphMamba = LegacyGraphMamba
|
357 |
QuantumEnhancedGraphMamba = LegacyGraphMamba
|
358 |
create_regularized_config = create_astrocyte_config
|