所有case的baseline，amd版本的ground truth都上传了

2026-02-08 09:42:14 +00:00
parent fffc5a9956
commit e6c55a648c
53 changed files with 3260 additions and 10 deletions
--- a/src/unifolm_wma/modules/attention.py
+++ b/src/unifolm_wma/modules/attention.py
@@ -125,7 +125,7 @@ class CrossAttention(nn.Module):
        context = default(context, x)

        if self.image_cross_attention and not spatial_self_attn:
-            assert 1 > 2, ">>> ERROR: should setup xformers and use efficient_forward ..."
+            # assert 1 > 2, ">>> ERROR: should setup xformers and use efficient_forward ..."
            context_agent_state = context[:, :self.agent_state_context_len, :]
            context_agent_action = context[:,
                                           self.agent_state_context_len:self.