[terminal_command] TERMINAL ()
 2 files changed, 24 insertions(+), 17 deletions(-)[mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine] $ git pushEnumerating objects: 11, done.Counting objects: 100% (11/11), done.Delta compression using up to 64 threadsCompressing objects: 100% (6/6), done.Writing objects: 100% (6/6), 708 bytes | 354.00 KiB/s, done.Total 6 (delta 5), reused 0 (delta 0), pack-reused 0 (from 0)remote: Resolving deltas: 100% (5/5), completed with 5 local objects.To github.com:p-doom/jasmine.git   4d6116c..0b08747  chore/cleanup-diffusion-branch -> chore/cleanup-diffusion-branch[mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine] $ queue[mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine] $ source .venv/bin/activate(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine] $ sbatch slurm/jobs/mihir/berlin/diffusion/breakout/dyn_dit_breakout_dev.shSubmitted batch job 32889(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine] $ queue(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine] $ scancel 32889(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine] $ git sttausgit: 'sttaus' is not a git command. See 'git --help'.The most similar command is        status(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine] $ git statusOn branch chore/cleanup-diffusion-branchYour branch is up to date with 'origin/chore/cleanup-diffusion-branch'.Changes not staged for commit:  (use "git add <file>..." to update what will be committed)  (use "git restore <file>..." to discard changes in working directory)        modified:   jasmine/genie.pyUntracked files:  (use "git add <file>..." to include in what will be committed)        data/_vizdoom.ini        data/jasmine_data/ViZDoomPPO/        data/uv.lock        jasmine/genie_bak.py        jasmine/genie_sliding.py        jasmine/sample_sliding.py        jasmine/test_dit.py        slurm/        uv.lockno changes added to commit (use "git add" and/or "git commit -a")(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine] $ git diffdiff --git a/jasmine/genie.py b/jasmine/genie.pyindex 2e64d24..5aa56f3 100644--- a/jasmine/genie.py+++ b/jasmine/genie.py@@ -648,6 +648,7 @@ class Genie(nnx.Module):         pad_shape = (B, seq_len - T, N, L)         pad = jax.random.normal(_rng_noise_full, pad_shape)         token_latents_BSNL = jnp.concatenate([token_latents_BTNL, pad], axis=1)+        dynamics_state = nnx.state(self.dynamics)          if self.use_gt_actions:             assert self.action_embed is not None@@ -676,6 +677,23 @@ class Genie(nnx.Module):             latents_BSNL, frame_i, rng = carry             rng, _rng_noise_context = jax.random.split(rng) +            # We need to reconstruct the submodule inside scan body to prevent trace context mismatches+            dynamics_diffusion = DynamicsDiffusion(+                model_dim=self.dyna_dim,+                ffn_dim=self.dyna_ffn_dim,+                latent_patch_dim=self.latent_patch_dim,+                latent_action_dim=self.latent_action_dim,+                num_blocks=self.dyna_num_blocks,+                num_heads=self.dyna_num_heads,+                denoise_steps=self.diffusion_denoise_steps,+                dropout=self.dropout,+                param_dtype=self.param_dtype,+                dtype=self.dtype,+                use_flash_attention=self.use_flash_attention,+                rngs=nnx.Rngs(0),+            )+            nnx.update(dynamics_diffusion, dynamics_state)+(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine] $ git commit -am "unfortunately we need to reconstruct the submodule inside scan body to prevent trace context mismatches during sampling"black....................................................................Passed[chore/cleanup-diffusion-branch 71deeb9] unfortunately we need to reconstruct the submodule inside scan body to prevent trace context mismatches during sampling 1 file changed, 19 insertions(+), 1 deletion(-)(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine] $ git pushEnumerating objects: 7, done.Counting objects: 100% (7/7), done.Delta compression using up to 64 threadsCompressing objects: 100% (4/4), done.Writing objects: 100% (4/4), 492 bytes | 246.00 KiB/s, done.Total 4 (delta 3), reused 0 (delta 0), pack-reused 0 (from 0)remote: Resolving deltas: 100% (3/3), completed with 3 local objects.To github.com:p-doom/jasmine.git   ad529f6..71deeb9  chore/cleanup-diffusion-branch -> chore/cleanup-diffusion-branch(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine] $ runner(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine_runner_1] $ git pullremote: Enumerating objects: 14, done.remote: Counting objects: 100% (14/14), done.remote: Compressing objects: 100% (3/3), done.remote: Total 8 (delta 5), reused 8 (delta 5), pack-reused 0 (from 0)Unpacking objects: 100% (8/8), 5.88 KiB | 103.00 KiB/s, done.From github.com:p-doom/jasmine   ad529f6..71deeb9  chore/cleanup-diffusion-branch -> origin/chore/cleanup-diffusion-branch   46cc9f5..7a461c2  positional_encoding_per_block  -> origin/positional_encoding_per_blockUpdating ad529f6..71deeb9Fast-forward jasmine/genie.py | 20 +++++++++++++++++++- 1 file changed, 19 insertions(+), 1 deletion(-)(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine_runner_1] $ cd slurm/(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine_runner_1/slurm] $ git pullAlready up to date.(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine_runner_1/slurm] $ cd ..(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine_runner_1] $ sbatch slurm/jobs/mihir/berlin/diffusion/coinrun/dyn_dit_coinrun.shSubmitted batch job 32890(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine_runner_1] $ sbatch slurm/jobs/mihir/berlin/diffusion/coinrun/ablations/dyn_dit_coinrun_8_steps.shSubmitted batch job 32891(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine_runner_1] $ sbatch slurm/jobs/mihir/berlin/diffusion/coinrun/ablations/dyn_dit_coinrun_16_steps.shSubmitted batch job 32892(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine_runner_1] $ sbatch slurm/jobs/mihir/berlin/diffusion/coinrun/ablations/dyn_dit_coinrun_32_steps.shSubmitted batch job 32893(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine_runner_1] $ sbatch slurm/jobs/mihir/berlin/diffusion/coinrun/ablations/dyn_dit_coinrun_64_steps.shSubmitted batch job 32894(jasmine) [mihir.mahajan@hai-login2.haicore.berlin:~/Projects/jasmine_runner_1] $ queue