p-doom/crowd-code-dataset-0.2 huggingface dataset:
[tab] test.ipynb (python)
data_dir="/home/franz.srambical/jafar/data_arrayrecord/dummy"
ckpt_dir="/home/franz.srambical/jafar/checkpoints/causal_dynamics_openai_grain_tok_restore"

%run sample.py \
    --dyna_type "causal" \
    --batch_size 1 \
    --seq_len 2 \
    --start_frame 1 \
    --checkpoint $ckpt_dir \
    --data_dir $data_dir