GeoTransolver: Fix attention and turn off feature broadcasting.

examples/cfd/external_aerodynamics/transformer_models/src/conf/data/volume.yaml

@@ -27,4 +27,5 @@ data_keys:
   - "stl_faces"
   - "stl_centers"
   - "stl_coordinates"
-
+  - "air_density"
+  - "stream_velocity"

examples/cfd/external_aerodynamics/transformer_models/src/conf/geotransolver_surface.yaml

@@ -39,7 +39,7 @@ data:
   include_sdf: false
   include_geometry: true
   geometry_sampling: 300_000
-  broadcast_global_features: true
+  broadcast_global_features: false
 
 
 # Logging configuration

examples/cfd/external_aerodynamics/transformer_models/src/inference_on_vtk.py

@@ -304,10 +304,10 @@ def build_data_dict(
 
     # Add flow parameters
     data_dict["air_density"] = torch.tensor(
-        [air_density], device=device, dtype=torch.float32
+        air_density, device=device, dtype=torch.float32
     )
     data_dict["stream_velocity"] = torch.tensor(
-        [stream_velocity], device=device, dtype=torch.float32
+        stream_velocity, device=device, dtype=torch.float32
     )
 
     return data_dict

physicsnemo/experimental/models/geotransolver/gale.py

@@ -457,7 +457,7 @@ def forward(
         attn = self.Attn(tuple(normed_inputs), global_context)
 
         # Residual connection after attention
-        fx_out = [attn[i] + normed_inputs[i] for i in range(len(normed_inputs))]
+        fx_out = [attn[i] + fx[i] for i in range(len(fx))]
 
         # Feed-forward network with residual connection
         fx_out = [self.ln_mlp1(_fx) + _fx for _fx in fx_out]