Upload folder using huggingface_hub

.gitattributes

@@ -492,3 +492,9 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 2026.04.23/15.32.08_train_llm_diffusion_unet_lowdim_box-close-v2/wandb/run-20260423_153304-it8p5fpg/run-it8p5fpg.wandb filter=lfs diff=lfs merge=lfs -text
 2026.04.24/19.32.45_train_llm_diffusion_unet_lowdim_box-close-v2/logs.json.txt filter=lfs diff=lfs merge=lfs -text
 2026.04.24/19.32.45_train_llm_diffusion_unet_lowdim_box-close-v2/wandb/run-20260424_193342-4hpop9xi/run-4hpop9xi.wandb filter=lfs diff=lfs merge=lfs -text
+2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/logs.json.txt filter=lfs diff=lfs merge=lfs -text
+2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/wandb/run-20260425_035339-ivkyq15y/run-ivkyq15y.wandb filter=lfs diff=lfs merge=lfs -text
+2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/logs.json.txt filter=lfs diff=lfs merge=lfs -text
+2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/wandb/run-20260425_035342-iafad9fe/run-iafad9fe.wandb filter=lfs diff=lfs merge=lfs -text
+2026.04.25/22.26.21_train_llm_diffusion_unet_lowdim_box-close-v2/logs.json.txt filter=lfs diff=lfs merge=lfs -text
+2026.04.25/22.26.21_train_llm_diffusion_unet_lowdim_box-close-v2/wandb/run-20260425_222718-n1fpkzkm/run-n1fpkzkm.wandb filter=lfs diff=lfs merge=lfs -text

2026.04.25/03.47.13_train_llm_diffusion_unet_lowdim_sweep-v2/.hydra/config.yaml

@@ -0,0 +1,193 @@
+name: train_llm_diffusion_unet_lowdim
+_target_: llmbc.workspace.train_llm_diffusion_unet_lowdim_workspace.TrainLLMDiffusionUnetLowdimWorkspace
+obs_dim: ${task.obs_dim}
+action_dim: ${task.action_dim}
+task_name: ${task.name}
+exp_name: default
+model_name: ${llm.name}
+horizon: 1
+n_obs_steps: 1
+n_action_steps: 1
+n_latency_steps: 0
+past_action_visible: false
+keypoint_visible_rate: 1.0
+obs_as_local_cond: false
+obs_as_global_cond: true
+pred_action_steps_only: false
+llm_orig_expert_feedback: true
+llm_do_sample: false
+policy:
+  _target_: llmbc.policy.llm_diffusion_unet_lowdim_policy.LLMDiffusionUnetLowdimPolicy
+  model:
+    _target_: llmbc.model.diffusion.conditional_unet1d.ConditionalUnet1D
+    input_dim: '${eval: ${task.action_dim} if ${obs_as_local_cond} or ${obs_as_global_cond}
+      else ${task.obs_dim} + ${task.action_dim}}'
+    local_cond_dim: '${eval: ${task.obs_dim} if ${obs_as_local_cond} else None}'
+    global_cond_dim: '${eval: ${task.obs_dim}*${n_obs_steps} if ${obs_as_global_cond}
+      else None}'
+    diffusion_step_embed_dim: 256
+    down_dims:
+    - 256
+    - 512
+    - 1024
+    kernel_size: 5
+    n_groups: 8
+    cond_predict_scale: true
+  noise_scheduler:
+    _target_: diffusers.schedulers.scheduling_ddpm.DDPMScheduler
+    num_train_timesteps: 100
+    beta_start: 0.0001
+    beta_end: 0.02
+    beta_schedule: squaredcos_cap_v2
+    variance_type: fixed_small
+    clip_sample: true
+    prediction_type: epsilon
+  horizon: ${horizon}
+  obs_dim: ${obs_dim}
+  action_dim: ${action_dim}
+  n_action_steps: ${eval:'${n_action_steps}+${n_latency_steps}'}
+  n_obs_steps: ${n_obs_steps}
+  num_inference_steps: 100
+  obs_as_local_cond: ${obs_as_local_cond}
+  obs_as_global_cond: ${obs_as_global_cond}
+  pred_action_steps_only: ${pred_action_steps_only}
+  oa_step_convention: true
+  llm_discriminator:
+    _target_: llmbc.discriminator.llm_ce_discriminator.LLMCEDiscriminator
+    task_id: ${task_name}
+    llm_translator:
+      _target_: llmbc.translator.llm_translator.LLMTranslator
+      cfg: ${llm}
+      obs_dim: ${obs_dim}
+      action_dim: ${action_dim}
+      horizon: ${horizon}
+      n_obs_steps: ${n_obs_steps}
+      n_action_steps: ${eval:'${n_action_steps}+${n_latency_steps}'}
+  loss_dp_weight: 1.0
+  loss_llm_weight: 0.001
+  normalize_llm_loss: true
+  reweight_llm_loss: true
+ema:
+  _target_: llmbc.model.diffusion.ema_model.EMAModel
+  update_after_step: 0
+  inv_gamma: 1.0
+  power: 0.75
+  min_value: 0.0
+  max_value: 0.9999
+dataloader:
+  batch_size: 16
+  num_workers: 0
+  shuffle: true
+  pin_memory: false
+  persistent_workers: false
+val_dataloader:
+  batch_size: 16
+  num_workers: 0
+  shuffle: false
+  pin_memory: false
+  persistent_workers: false
+optimizer:
+  _target_: torch.optim.AdamW
+  lr: 0.0001
+  betas:
+  - 0.95
+  - 0.999
+  eps: 1.0e-08
+  weight_decay: 1.0e-06
+training:
+  device: cuda:0
+  seed: 42
+  debug: false
+  resume: true
+  lr_scheduler: cosine
+  lr_warmup_steps: 500
+  num_epochs: 1001
+  gradient_accumulate_every: 16
+  use_ema: true
+  rollout_every: 5
+  checkpoint_every: 5
+  val_every: 1
+  sample_every: 5
+  max_train_steps: null
+  max_val_steps: null
+  tqdm_interval_sec: 1.0
+logging:
+  project: ${task.name}-training-mix
+  resume: true
+  mode: online
+  name: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
+  tags:
+  - ${name}
+  - ${task_name}
+  - ${exp_name}
+  id: null
+  group: null
+checkpoint:
+  topk:
+    monitor_key: test_success_rate
+    mode: max
+    k: 5
+    format_str: epoch={epoch:04d}-test_success_rate={test_success_rate:.3f}.ckpt
+  save_last_ckpt: true
+  save_last_snapshot: false
+multi_run:
+  run_dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+  wandb_name_base: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
+task:
+  name: sweep-v2
+  obs_dim: 9
+  action_dim: 4
+  env_runner:
+    _target_: llmbc.env_runner.metaworld_lowdim_runner.MetaworldLowdimRunner
+    env_name: llf-metaworld-sweep-v2
+    n_train: 10
+    n_test: 50
+    n_envs: 10
+    max_steps: 30
+    n_obs_steps: ${n_obs_steps}
+    n_action_steps: ${n_action_steps}
+    instruction_type: b
+    feedback_type:
+    - hp
+    - hn
+    - fp
+    visual: false
+    discount: 0.9
+  dataset:
+    _target_: llmbc.dataset.metaworld_lowdim_dataset.MetaworldLowdimDataset
+    data_path: datasets/sweep-v2-mix(csp3).pt
+    data_path2: datasets/sweep-v2.pt
+    horizon: ${horizon}
+    pad_before: ${eval:'${n_obs_steps}-1'}
+    pad_after: ${eval:'${n_action_steps}-1'}
+    obs_eef_target: true
+    use_manual_normalizer: false
+    val_ratio: 0.1
+    dummy_normalizer: false
+  instructor:
+    _target_: llmbc.translator.instructor.metaworld_instructor.sweep_v2_instructor.SweepV2Instructor
+llm:
+  name: HuggingFaceTB/SmolLM2-135M-Instruct
+  model_name: SmolLM2-135M-Instruct
+  config_target: llmbc.model.llm.llama_lowdim_model.LowdimLlamaConfig
+  causal_lm_target: llmbc.model.llm.llama_lowdim_model.LowdimLlamaForCausalLM
+  use_quantization: false
+  use_joint_mlp_projector: true
+  llm_mode: ete-finetuned
+  finetune_mode: orig
+  checkpoint: data/outputs/2025.11.01/23.03.01_train_llm_lowdim_sweep-v2/HuggingFaceTB/SmolLM2-135M-Instruct-finetuned-sweep-v2/checkpoint-5690
+  max_length: 100
+  lora_config:
+    r: 32
+    lora_alpha: 64
+    lora_dropout: 0.05
+    bias: none
+    task_type: CAUSAL_LM
+  prompter:
+    _target_: llmbc.translator.prompter.smollm2_prompter.SmolLM2Prompter
+    use_joint_mlp_projector: true
+  hydra:
+    job:
+      override_dirname: ${model_name}
+    run:
+      dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${model_name}

2026.04.25/03.47.13_train_llm_diffusion_unet_lowdim_sweep-v2/.hydra/hydra.yaml

@@ -0,0 +1,155 @@
+hydra:
+  run:
+    dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+  sweep:
+    dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+    subdir: ${hydra.job.num}
+  launcher:
+    _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
+  sweeper:
+    _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
+    max_batch_size: null
+    params: null
+  help:
+    app_name: ${hydra.job.name}
+    header: '${hydra.help.app_name} is powered by Hydra.
+
+      '
+    footer: 'Powered by Hydra (https://hydra.cc)
+
+      Use --hydra-help to view Hydra specific help
+
+      '
+    template: '${hydra.help.header}
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (group=option)
+
+
+      $APP_CONFIG_GROUPS
+
+
+      == Config ==
+
+      Override anything in the config (foo.bar=value)
+
+
+      $CONFIG
+
+
+      ${hydra.help.footer}
+
+      '
+  hydra_help:
+    template: 'Hydra (${hydra.runtime.version})
+
+      See https://hydra.cc for more info.
+
+
+      == Flags ==
+
+      $FLAGS_HELP
+
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (For example, append hydra/job_logging=disabled
+      to command line)
+
+
+      $HYDRA_CONFIG_GROUPS
+
+
+      Use ''--cfg hydra'' to Show the Hydra config.
+
+      '
+    hydra_help: ???
+  hydra_logging:
+    version: 1
+    formatters:
+      simple:
+        format: '[%(asctime)s][HYDRA] %(message)s'
+    handlers:
+      console:
+        class: logging.StreamHandler
+        formatter: simple
+        stream: ext://sys.stdout
+    root:
+      level: INFO
+      handlers:
+      - console
+    loggers:
+      logging_example:
+        level: DEBUG
+    disable_existing_loggers: false
+  job_logging:
+    version: 1
+    formatters:
+      simple:
+        format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
+    handlers:
+      console:
+        class: logging.StreamHandler
+        formatter: simple
+        stream: ext://sys.stdout
+      file:
+        class: logging.FileHandler
+        formatter: simple
+        filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
+    root:
+      level: INFO
+      handlers:
+      - console
+      - file
+    disable_existing_loggers: false
+  env: {}
+  mode: RUN
+  searchpath: []
+  callbacks: {}
+  output_subdir: .hydra
+  overrides:
+    hydra:
+    - hydra.mode=RUN
+    task:
+    - training.seed=42
+  job:
+    name: train
+    chdir: null
+    override_dirname: training.seed=42
+    id: ???
+    num: ???
+    config_name: llmdp_sweep-v2_mix(csp3).yaml
+    env_set: {}
+    env_copy: []
+    config:
+      override_dirname:
+        kv_sep: '='
+        item_sep: ','
+        exclude_keys: []
+  runtime:
+    version: 1.2.0
+    version_base: '1.2'
+    cwd: /workspace/LLM-BC
+    config_sources:
+    - path: hydra.conf
+      schema: pkg
+      provider: hydra
+    - path: /workspace/LLM-BC/config/mix_data
+      schema: file
+      provider: main
+    - path: ''
+      schema: structured
+      provider: schema
+    output_dir: /workspace/LLM-BC/data/outputs/2026.04.25/03.47.13_train_llm_diffusion_unet_lowdim_sweep-v2
+    choices:
+      hydra/env: default
+      hydra/callbacks: null
+      hydra/job_logging: default
+      hydra/hydra_logging: default
+      hydra/hydra_help: default
+      hydra/help: default
+      hydra/sweeper: basic
+      hydra/launcher: basic
+      hydra/output: default
+  verbose: false

2026.04.25/03.47.13_train_llm_diffusion_unet_lowdim_sweep-v2/.hydra/overrides.yaml

@@ -0,0 +1 @@
+- training.seed=42

2026.04.25/03.47.13_train_llm_diffusion_unet_lowdim_sweep-v2/train.log

@@ -0,0 +1,7 @@
+[2026-04-25 03:47:15,107][numexpr.utils][INFO] - Note: detected 144 virtual cores but NumExpr set to maximum of 64, check "NUMEXPR_MAX_THREADS" environment variable.
+[2026-04-25 03:47:15,107][numexpr.utils][INFO] - Note: NumExpr detected 144 cores but "NUMEXPR_MAX_THREADS" not set, so enforcing safe limit of 16.
+[2026-04-25 03:47:15,107][numexpr.utils][INFO] - NumExpr defaulting to 16 threads.
+[2026-04-25 03:47:17,304][datasets][INFO] - PyTorch version 2.2.2 available.
+[2026-04-25 03:47:17,305][datasets][INFO] - TensorFlow version 2.15.1 available.
+[2026-04-25 03:47:17,305][datasets][INFO] - JAX version 0.4.30 available.
+[2026-04-25 03:47:19,027][llmbc.model.diffusion.conditional_unet1d][INFO] - number of parameters: 6.501479e+07

2026.04.25/03.47.19_train_llm_diffusion_unet_lowdim_sweep-v2/.hydra/config.yaml

@@ -0,0 +1,193 @@
+name: train_llm_diffusion_unet_lowdim
+_target_: llmbc.workspace.train_llm_diffusion_unet_lowdim_workspace.TrainLLMDiffusionUnetLowdimWorkspace
+obs_dim: ${task.obs_dim}
+action_dim: ${task.action_dim}
+task_name: ${task.name}
+exp_name: default
+model_name: ${llm.name}
+horizon: 1
+n_obs_steps: 1
+n_action_steps: 1
+n_latency_steps: 0
+past_action_visible: false
+keypoint_visible_rate: 1.0
+obs_as_local_cond: false
+obs_as_global_cond: true
+pred_action_steps_only: false
+llm_orig_expert_feedback: true
+llm_do_sample: false
+policy:
+  _target_: llmbc.policy.llm_diffusion_unet_lowdim_policy.LLMDiffusionUnetLowdimPolicy
+  model:
+    _target_: llmbc.model.diffusion.conditional_unet1d.ConditionalUnet1D
+    input_dim: '${eval: ${task.action_dim} if ${obs_as_local_cond} or ${obs_as_global_cond}
+      else ${task.obs_dim} + ${task.action_dim}}'
+    local_cond_dim: '${eval: ${task.obs_dim} if ${obs_as_local_cond} else None}'
+    global_cond_dim: '${eval: ${task.obs_dim}*${n_obs_steps} if ${obs_as_global_cond}
+      else None}'
+    diffusion_step_embed_dim: 256
+    down_dims:
+    - 256
+    - 512
+    - 1024
+    kernel_size: 5
+    n_groups: 8
+    cond_predict_scale: true
+  noise_scheduler:
+    _target_: diffusers.schedulers.scheduling_ddpm.DDPMScheduler
+    num_train_timesteps: 100
+    beta_start: 0.0001
+    beta_end: 0.02
+    beta_schedule: squaredcos_cap_v2
+    variance_type: fixed_small
+    clip_sample: true
+    prediction_type: epsilon
+  horizon: ${horizon}
+  obs_dim: ${obs_dim}
+  action_dim: ${action_dim}
+  n_action_steps: ${eval:'${n_action_steps}+${n_latency_steps}'}
+  n_obs_steps: ${n_obs_steps}
+  num_inference_steps: 100
+  obs_as_local_cond: ${obs_as_local_cond}
+  obs_as_global_cond: ${obs_as_global_cond}
+  pred_action_steps_only: ${pred_action_steps_only}
+  oa_step_convention: true
+  llm_discriminator:
+    _target_: llmbc.discriminator.llm_ce_discriminator.LLMCEDiscriminator
+    task_id: ${task_name}
+    llm_translator:
+      _target_: llmbc.translator.llm_translator.LLMTranslator
+      cfg: ${llm}
+      obs_dim: ${obs_dim}
+      action_dim: ${action_dim}
+      horizon: ${horizon}
+      n_obs_steps: ${n_obs_steps}
+      n_action_steps: ${eval:'${n_action_steps}+${n_latency_steps}'}
+  loss_dp_weight: 1.0
+  loss_llm_weight: 0.001
+  normalize_llm_loss: true
+  reweight_llm_loss: true
+ema:
+  _target_: llmbc.model.diffusion.ema_model.EMAModel
+  update_after_step: 0
+  inv_gamma: 1.0
+  power: 0.75
+  min_value: 0.0
+  max_value: 0.9999
+dataloader:
+  batch_size: 16
+  num_workers: 0
+  shuffle: true
+  pin_memory: false
+  persistent_workers: false
+val_dataloader:
+  batch_size: 16
+  num_workers: 0
+  shuffle: false
+  pin_memory: false
+  persistent_workers: false
+optimizer:
+  _target_: torch.optim.AdamW
+  lr: 0.0001
+  betas:
+  - 0.95
+  - 0.999
+  eps: 1.0e-08
+  weight_decay: 1.0e-06
+training:
+  device: cuda:0
+  seed: 43
+  debug: false
+  resume: true
+  lr_scheduler: cosine
+  lr_warmup_steps: 500
+  num_epochs: 1001
+  gradient_accumulate_every: 16
+  use_ema: true
+  rollout_every: 5
+  checkpoint_every: 5
+  val_every: 1
+  sample_every: 5
+  max_train_steps: null
+  max_val_steps: null
+  tqdm_interval_sec: 1.0
+logging:
+  project: ${task.name}-training-mix
+  resume: true
+  mode: online
+  name: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
+  tags:
+  - ${name}
+  - ${task_name}
+  - ${exp_name}
+  id: null
+  group: null
+checkpoint:
+  topk:
+    monitor_key: test_success_rate
+    mode: max
+    k: 5
+    format_str: epoch={epoch:04d}-test_success_rate={test_success_rate:.3f}.ckpt
+  save_last_ckpt: true
+  save_last_snapshot: false
+multi_run:
+  run_dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+  wandb_name_base: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
+task:
+  name: sweep-v2
+  obs_dim: 9
+  action_dim: 4
+  env_runner:
+    _target_: llmbc.env_runner.metaworld_lowdim_runner.MetaworldLowdimRunner
+    env_name: llf-metaworld-sweep-v2
+    n_train: 10
+    n_test: 50
+    n_envs: 10
+    max_steps: 30
+    n_obs_steps: ${n_obs_steps}
+    n_action_steps: ${n_action_steps}
+    instruction_type: b
+    feedback_type:
+    - hp
+    - hn
+    - fp
+    visual: false
+    discount: 0.9
+  dataset:
+    _target_: llmbc.dataset.metaworld_lowdim_dataset.MetaworldLowdimDataset
+    data_path: datasets/sweep-v2-mix(csp3).pt
+    data_path2: datasets/sweep-v2.pt
+    horizon: ${horizon}
+    pad_before: ${eval:'${n_obs_steps}-1'}
+    pad_after: ${eval:'${n_action_steps}-1'}
+    obs_eef_target: true
+    use_manual_normalizer: false
+    val_ratio: 0.1
+    dummy_normalizer: false
+  instructor:
+    _target_: llmbc.translator.instructor.metaworld_instructor.sweep_v2_instructor.SweepV2Instructor
+llm:
+  name: HuggingFaceTB/SmolLM2-135M-Instruct
+  model_name: SmolLM2-135M-Instruct
+  config_target: llmbc.model.llm.llama_lowdim_model.LowdimLlamaConfig
+  causal_lm_target: llmbc.model.llm.llama_lowdim_model.LowdimLlamaForCausalLM
+  use_quantization: false
+  use_joint_mlp_projector: true
+  llm_mode: ete-finetuned
+  finetune_mode: orig
+  checkpoint: data/outputs/2025.11.01/23.03.01_train_llm_lowdim_sweep-v2/HuggingFaceTB/SmolLM2-135M-Instruct-finetuned-sweep-v2/checkpoint-5690
+  max_length: 100
+  lora_config:
+    r: 32
+    lora_alpha: 64
+    lora_dropout: 0.05
+    bias: none
+    task_type: CAUSAL_LM
+  prompter:
+    _target_: llmbc.translator.prompter.smollm2_prompter.SmolLM2Prompter
+    use_joint_mlp_projector: true
+  hydra:
+    job:
+      override_dirname: ${model_name}
+    run:
+      dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${model_name}

2026.04.25/03.47.19_train_llm_diffusion_unet_lowdim_sweep-v2/.hydra/hydra.yaml

@@ -0,0 +1,155 @@
+hydra:
+  run:
+    dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+  sweep:
+    dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+    subdir: ${hydra.job.num}
+  launcher:
+    _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
+  sweeper:
+    _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
+    max_batch_size: null
+    params: null
+  help:
+    app_name: ${hydra.job.name}
+    header: '${hydra.help.app_name} is powered by Hydra.
+
+      '
+    footer: 'Powered by Hydra (https://hydra.cc)
+
+      Use --hydra-help to view Hydra specific help
+
+      '
+    template: '${hydra.help.header}
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (group=option)
+
+
+      $APP_CONFIG_GROUPS
+
+
+      == Config ==
+
+      Override anything in the config (foo.bar=value)
+
+
+      $CONFIG
+
+
+      ${hydra.help.footer}
+
+      '
+  hydra_help:
+    template: 'Hydra (${hydra.runtime.version})
+
+      See https://hydra.cc for more info.
+
+
+      == Flags ==
+
+      $FLAGS_HELP
+
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (For example, append hydra/job_logging=disabled
+      to command line)
+
+
+      $HYDRA_CONFIG_GROUPS
+
+
+      Use ''--cfg hydra'' to Show the Hydra config.
+
+      '
+    hydra_help: ???
+  hydra_logging:
+    version: 1
+    formatters:
+      simple:
+        format: '[%(asctime)s][HYDRA] %(message)s'
+    handlers:
+      console:
+        class: logging.StreamHandler
+        formatter: simple
+        stream: ext://sys.stdout
+    root:
+      level: INFO
+      handlers:
+      - console
+    loggers:
+      logging_example:
+        level: DEBUG
+    disable_existing_loggers: false
+  job_logging:
+    version: 1
+    formatters:
+      simple:
+        format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
+    handlers:
+      console:
+        class: logging.StreamHandler
+        formatter: simple
+        stream: ext://sys.stdout
+      file:
+        class: logging.FileHandler
+        formatter: simple
+        filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
+    root:
+      level: INFO
+      handlers:
+      - console
+      - file
+    disable_existing_loggers: false
+  env: {}
+  mode: RUN
+  searchpath: []
+  callbacks: {}
+  output_subdir: .hydra
+  overrides:
+    hydra:
+    - hydra.mode=RUN
+    task:
+    - training.seed=43
+  job:
+    name: train
+    chdir: null
+    override_dirname: training.seed=43
+    id: ???
+    num: ???
+    config_name: llmdp_sweep-v2_mix(csp3).yaml
+    env_set: {}
+    env_copy: []
+    config:
+      override_dirname:
+        kv_sep: '='
+        item_sep: ','
+        exclude_keys: []
+  runtime:
+    version: 1.2.0
+    version_base: '1.2'
+    cwd: /workspace/LLM-BC
+    config_sources:
+    - path: hydra.conf
+      schema: pkg
+      provider: hydra
+    - path: /workspace/LLM-BC/config/mix_data
+      schema: file
+      provider: main
+    - path: ''
+      schema: structured
+      provider: schema
+    output_dir: /workspace/LLM-BC/data/outputs/2026.04.25/03.47.19_train_llm_diffusion_unet_lowdim_sweep-v2
+    choices:
+      hydra/env: default
+      hydra/callbacks: null
+      hydra/job_logging: default
+      hydra/hydra_logging: default
+      hydra/hydra_help: default
+      hydra/help: default
+      hydra/sweeper: basic
+      hydra/launcher: basic
+      hydra/output: default
+  verbose: false

2026.04.25/03.47.19_train_llm_diffusion_unet_lowdim_sweep-v2/.hydra/overrides.yaml

@@ -0,0 +1 @@
+- training.seed=43

2026.04.25/03.47.19_train_llm_diffusion_unet_lowdim_sweep-v2/train.log

@@ -0,0 +1,7 @@
+[2026-04-25 03:47:21,387][numexpr.utils][INFO] - Note: detected 144 virtual cores but NumExpr set to maximum of 64, check "NUMEXPR_MAX_THREADS" environment variable.
+[2026-04-25 03:47:21,387][numexpr.utils][INFO] - Note: NumExpr detected 144 cores but "NUMEXPR_MAX_THREADS" not set, so enforcing safe limit of 16.
+[2026-04-25 03:47:21,387][numexpr.utils][INFO] - NumExpr defaulting to 16 threads.
+[2026-04-25 03:47:23,695][datasets][INFO] - PyTorch version 2.2.2 available.
+[2026-04-25 03:47:23,696][datasets][INFO] - TensorFlow version 2.15.1 available.
+[2026-04-25 03:47:23,696][datasets][INFO] - JAX version 0.4.30 available.
+[2026-04-25 03:47:25,356][llmbc.model.diffusion.conditional_unet1d][INFO] - number of parameters: 6.501479e+07

2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/.hydra/config.yaml

@@ -0,0 +1,193 @@
+name: train_llm_diffusion_unet_lowdim
+_target_: llmbc.workspace.train_llm_diffusion_unet_lowdim_workspace.TrainLLMDiffusionUnetLowdimWorkspace
+obs_dim: ${task.obs_dim}
+action_dim: ${task.action_dim}
+task_name: ${task.name}
+exp_name: default
+model_name: ${llm.name}
+horizon: 1
+n_obs_steps: 1
+n_action_steps: 1
+n_latency_steps: 0
+past_action_visible: false
+keypoint_visible_rate: 1.0
+obs_as_local_cond: false
+obs_as_global_cond: true
+pred_action_steps_only: false
+llm_orig_expert_feedback: true
+llm_do_sample: false
+policy:
+  _target_: llmbc.policy.llm_diffusion_unet_lowdim_policy.LLMDiffusionUnetLowdimPolicy
+  model:
+    _target_: llmbc.model.diffusion.conditional_unet1d.ConditionalUnet1D
+    input_dim: '${eval: ${task.action_dim} if ${obs_as_local_cond} or ${obs_as_global_cond}
+      else ${task.obs_dim} + ${task.action_dim}}'
+    local_cond_dim: '${eval: ${task.obs_dim} if ${obs_as_local_cond} else None}'
+    global_cond_dim: '${eval: ${task.obs_dim}*${n_obs_steps} if ${obs_as_global_cond}
+      else None}'
+    diffusion_step_embed_dim: 256
+    down_dims:
+    - 256
+    - 512
+    - 1024
+    kernel_size: 5
+    n_groups: 8
+    cond_predict_scale: true
+  noise_scheduler:
+    _target_: diffusers.schedulers.scheduling_ddpm.DDPMScheduler
+    num_train_timesteps: 100
+    beta_start: 0.0001
+    beta_end: 0.02
+    beta_schedule: squaredcos_cap_v2
+    variance_type: fixed_small
+    clip_sample: true
+    prediction_type: epsilon
+  horizon: ${horizon}
+  obs_dim: ${obs_dim}
+  action_dim: ${action_dim}
+  n_action_steps: ${eval:'${n_action_steps}+${n_latency_steps}'}
+  n_obs_steps: ${n_obs_steps}
+  num_inference_steps: 100
+  obs_as_local_cond: ${obs_as_local_cond}
+  obs_as_global_cond: ${obs_as_global_cond}
+  pred_action_steps_only: ${pred_action_steps_only}
+  oa_step_convention: true
+  llm_discriminator:
+    _target_: llmbc.discriminator.llm_ce_discriminator.LLMCEDiscriminator
+    task_id: ${task_name}
+    llm_translator:
+      _target_: llmbc.translator.llm_translator.LLMTranslator
+      cfg: ${llm}
+      obs_dim: ${obs_dim}
+      action_dim: ${action_dim}
+      horizon: ${horizon}
+      n_obs_steps: ${n_obs_steps}
+      n_action_steps: ${eval:'${n_action_steps}+${n_latency_steps}'}
+  loss_dp_weight: 1.0
+  loss_llm_weight: 0.001
+  normalize_llm_loss: true
+  reweight_llm_loss: true
+ema:
+  _target_: llmbc.model.diffusion.ema_model.EMAModel
+  update_after_step: 0
+  inv_gamma: 1.0
+  power: 0.75
+  min_value: 0.0
+  max_value: 0.9999
+dataloader:
+  batch_size: 16
+  num_workers: 0
+  shuffle: true
+  pin_memory: false
+  persistent_workers: false
+val_dataloader:
+  batch_size: 16
+  num_workers: 0
+  shuffle: false
+  pin_memory: false
+  persistent_workers: false
+optimizer:
+  _target_: torch.optim.AdamW
+  lr: 0.0001
+  betas:
+  - 0.95
+  - 0.999
+  eps: 1.0e-08
+  weight_decay: 1.0e-06
+training:
+  device: cuda:0
+  seed: 42
+  debug: false
+  resume: true
+  lr_scheduler: cosine
+  lr_warmup_steps: 500
+  num_epochs: 1001
+  gradient_accumulate_every: 16
+  use_ema: true
+  rollout_every: 5
+  checkpoint_every: 5
+  val_every: 1
+  sample_every: 5
+  max_train_steps: null
+  max_val_steps: null
+  tqdm_interval_sec: 1.0
+logging:
+  project: ${task.name}-training-mix
+  resume: true
+  mode: online
+  name: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
+  tags:
+  - ${name}
+  - ${task_name}
+  - ${exp_name}
+  id: null
+  group: null
+checkpoint:
+  topk:
+    monitor_key: test_success_rate
+    mode: max
+    k: 5
+    format_str: epoch={epoch:04d}-test_success_rate={test_success_rate:.3f}.ckpt
+  save_last_ckpt: true
+  save_last_snapshot: false
+multi_run:
+  run_dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+  wandb_name_base: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
+task:
+  name: sweep-v2
+  obs_dim: 9
+  action_dim: 4
+  env_runner:
+    _target_: llmbc.env_runner.metaworld_lowdim_runner.MetaworldLowdimRunner
+    env_name: llf-metaworld-sweep-v2
+    n_train: 10
+    n_test: 50
+    n_envs: 10
+    max_steps: 30
+    n_obs_steps: ${n_obs_steps}
+    n_action_steps: ${n_action_steps}
+    instruction_type: b
+    feedback_type:
+    - hp
+    - hn
+    - fp
+    visual: false
+    discount: 0.9
+  dataset:
+    _target_: llmbc.dataset.metaworld_lowdim_dataset.MetaworldLowdimDataset
+    data_path: datasets/sweep-v2-mix(csp3).pt
+    data_path2: datasets/sweep-v2.pt
+    horizon: ${horizon}
+    pad_before: ${eval:'${n_obs_steps}-1'}
+    pad_after: ${eval:'${n_action_steps}-1'}
+    obs_eef_target: true
+    use_manual_normalizer: false
+    val_ratio: 0.1
+    dummy_normalizer: false
+  instructor:
+    _target_: llmbc.translator.instructor.metaworld_instructor.sweep_v2_instructor.SweepV2Instructor
+llm:
+  name: HuggingFaceTB/SmolLM2-135M-Instruct
+  model_name: SmolLM2-135M-Instruct
+  config_target: llmbc.model.llm.llama_lowdim_model.LowdimLlamaConfig
+  causal_lm_target: llmbc.model.llm.llama_lowdim_model.LowdimLlamaForCausalLM
+  use_quantization: false
+  use_joint_mlp_projector: true
+  llm_mode: ete-finetuned
+  finetune_mode: orig
+  checkpoint: data/outputs/2025.11.01/23.03.01_train_llm_lowdim_sweep-v2/HuggingFaceTB/SmolLM2-135M-Instruct-finetuned-sweep-v2/checkpoint-5690
+  max_length: 100
+  lora_config:
+    r: 32
+    lora_alpha: 64
+    lora_dropout: 0.05
+    bias: none
+    task_type: CAUSAL_LM
+  prompter:
+    _target_: llmbc.translator.prompter.smollm2_prompter.SmolLM2Prompter
+    use_joint_mlp_projector: true
+  hydra:
+    job:
+      override_dirname: ${model_name}
+    run:
+      dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${model_name}

2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/.hydra/hydra.yaml

@@ -0,0 +1,155 @@
+hydra:
+  run:
+    dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+  sweep:
+    dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+    subdir: ${hydra.job.num}
+  launcher:
+    _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
+  sweeper:
+    _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
+    max_batch_size: null
+    params: null
+  help:
+    app_name: ${hydra.job.name}
+    header: '${hydra.help.app_name} is powered by Hydra.
+
+      '
+    footer: 'Powered by Hydra (https://hydra.cc)
+
+      Use --hydra-help to view Hydra specific help
+
+      '
+    template: '${hydra.help.header}
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (group=option)
+
+
+      $APP_CONFIG_GROUPS
+
+
+      == Config ==
+
+      Override anything in the config (foo.bar=value)
+
+
+      $CONFIG
+
+
+      ${hydra.help.footer}
+
+      '
+  hydra_help:
+    template: 'Hydra (${hydra.runtime.version})
+
+      See https://hydra.cc for more info.
+
+
+      == Flags ==
+
+      $FLAGS_HELP
+
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (For example, append hydra/job_logging=disabled
+      to command line)
+
+
+      $HYDRA_CONFIG_GROUPS
+
+
+      Use ''--cfg hydra'' to Show the Hydra config.
+
+      '
+    hydra_help: ???
+  hydra_logging:
+    version: 1
+    formatters:
+      simple:
+        format: '[%(asctime)s][HYDRA] %(message)s'
+    handlers:
+      console:
+        class: logging.StreamHandler
+        formatter: simple
+        stream: ext://sys.stdout
+    root:
+      level: INFO
+      handlers:
+      - console
+    loggers:
+      logging_example:
+        level: DEBUG
+    disable_existing_loggers: false
+  job_logging:
+    version: 1
+    formatters:
+      simple:
+        format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
+    handlers:
+      console:
+        class: logging.StreamHandler
+        formatter: simple
+        stream: ext://sys.stdout
+      file:
+        class: logging.FileHandler
+        formatter: simple
+        filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
+    root:
+      level: INFO
+      handlers:
+      - console
+      - file
+    disable_existing_loggers: false
+  env: {}
+  mode: RUN
+  searchpath: []
+  callbacks: {}
+  output_subdir: .hydra
+  overrides:
+    hydra:
+    - hydra.mode=RUN
+    task:
+    - training.seed=42
+  job:
+    name: train
+    chdir: null
+    override_dirname: training.seed=42
+    id: ???
+    num: ???
+    config_name: llmdp_sweep-v2_mix(csp3).yaml
+    env_set: {}
+    env_copy: []
+    config:
+      override_dirname:
+        kv_sep: '='
+        item_sep: ','
+        exclude_keys: []
+  runtime:
+    version: 1.2.0
+    version_base: '1.2'
+    cwd: /workspace/LLM-BC
+    config_sources:
+    - path: hydra.conf
+      schema: pkg
+      provider: hydra
+    - path: /workspace/LLM-BC/config/mix_data
+      schema: file
+      provider: main
+    - path: ''
+      schema: structured
+      provider: schema
+    output_dir: /workspace/LLM-BC/data/outputs/2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2
+    choices:
+      hydra/env: default
+      hydra/callbacks: null
+      hydra/job_logging: default
+      hydra/hydra_logging: default
+      hydra/hydra_help: default
+      hydra/help: default
+      hydra/sweeper: basic
+      hydra/launcher: basic
+      hydra/output: default
+  verbose: false

2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/.hydra/overrides.yaml

@@ -0,0 +1 @@
+- training.seed=42

2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/checkpoints/epoch=0180-test_success_rate=0.740.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9374a1bd4fa15f7d2831b4083fe9ace1a39357b51c6c1577e4cdb4af4132b707
+size 1040498994

2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/checkpoints/epoch=0265-test_success_rate=0.680.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:416fe9b5412b07a3c2bb3b02f63d7370858921112ee5e143d27859bb6452f253
+size 1040498994

2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/checkpoints/epoch=0410-test_success_rate=0.680.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c91a2faee47005f5e07b7306020afe42359a262fafbfb3418136cdd066725b77
+size 1040498994

2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/checkpoints/epoch=0455-test_success_rate=0.780.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3d535b2cf88fe4d3ceeb3cbe86c20d989a9e4c1c7a0681cb8876055709def17a
+size 1040498994

2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/checkpoints/epoch=0515-test_success_rate=0.680.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:623eb4c2f6bb2e3ab206cee4358793169e412fa7c610c6c1326555f3f77c432d
+size 1040498994

2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/checkpoints/latest.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d0814a9d001e4da156135f8654859113aae6d0f3694a623b9f2711a4ad36e3c0
+size 1040498994

2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/logs.json.txt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:22e8eca293b2a1177ac050839d717267ac006bb060e1c8cf456e518d0d76ccee
+size 120440567

2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/train.log

@@ -0,0 +1,14 @@
+[2026-04-25 03:52:04,619][numexpr.utils][INFO] - Note: detected 144 virtual cores but NumExpr set to maximum of 64, check "NUMEXPR_MAX_THREADS" environment variable.
+[2026-04-25 03:52:04,619][numexpr.utils][INFO] - Note: NumExpr detected 144 cores but "NUMEXPR_MAX_THREADS" not set, so enforcing safe limit of 16.
+[2026-04-25 03:52:04,619][numexpr.utils][INFO] - NumExpr defaulting to 16 threads.
+[2026-04-25 03:52:06,853][datasets][INFO] - PyTorch version 2.2.2 available.
+[2026-04-25 03:52:06,854][datasets][INFO] - TensorFlow version 2.15.1 available.
+[2026-04-25 03:52:06,855][datasets][INFO] - JAX version 0.4.30 available.
+[2026-04-25 03:52:08,560][llmbc.model.diffusion.conditional_unet1d][INFO] - number of parameters: 6.501479e+07
+[2026-04-25 03:52:12,545][root][INFO] - running build_ext
+[2026-04-25 03:52:12,547][root][INFO] - building 'mujoco_py.cymj' extension
+[2026-04-25 03:52:12,547][root][INFO] - gcc -pthread -B /venv/llm-bc/compiler_compat -Wno-unused-result -Wsign-compare -DNDEBUG -O2 -Wall -fPIC -O2 -isystem /venv/llm-bc/include -I/venv/llm-bc/include -fPIC -O2 -isystem /venv/llm-bc/include -fPIC -I/venv/llm-bc/lib/python3.9/site-packages/mujoco_py -I/root/.mujoco/mujoco210/include -I/venv/llm-bc/lib/python3.9/site-packages/numpy/core/include -I/venv/llm-bc/include/python3.9 -c /venv/llm-bc/lib/python3.9/site-packages/mujoco_py/cymj.c -o /venv/llm-bc/lib/python3.9/site-packages/mujoco_py/generated/_pyxbld_2.1.2.14_39_linuxcpuextensionbuilder/temp.linux-x86_64-cpython-39/venv/llm-bc/lib/python3.9/site-packages/mujoco_py/cymj.o -fopenmp -w
+[2026-04-25 03:52:55,118][root][INFO] - gcc -pthread -B /venv/llm-bc/compiler_compat -Wno-unused-result -Wsign-compare -DNDEBUG -O2 -Wall -fPIC -O2 -isystem /venv/llm-bc/include -I/venv/llm-bc/include -fPIC -O2 -isystem /venv/llm-bc/include -fPIC -I/venv/llm-bc/lib/python3.9/site-packages/mujoco_py -I/root/.mujoco/mujoco210/include -I/venv/llm-bc/lib/python3.9/site-packages/numpy/core/include -I/venv/llm-bc/include/python3.9 -c /venv/llm-bc/lib/python3.9/site-packages/mujoco_py/gl/osmesashim.c -o /venv/llm-bc/lib/python3.9/site-packages/mujoco_py/generated/_pyxbld_2.1.2.14_39_linuxcpuextensionbuilder/temp.linux-x86_64-cpython-39/venv/llm-bc/lib/python3.9/site-packages/mujoco_py/gl/osmesashim.o -fopenmp -w
+[2026-04-25 03:52:55,248][root][INFO] - gcc -pthread -B /venv/llm-bc/compiler_compat -shared -Wl,-rpath,/venv/llm-bc/lib -Wl,-rpath-link,/venv/llm-bc/lib -L/venv/llm-bc/lib -L/venv/llm-bc/lib -Wl,-rpath,/venv/llm-bc/lib -Wl,-rpath-link,/venv/llm-bc/lib -L/venv/llm-bc/lib /venv/llm-bc/lib/python3.9/site-packages/mujoco_py/generated/_pyxbld_2.1.2.14_39_linuxcpuextensionbuilder/temp.linux-x86_64-cpython-39/venv/llm-bc/lib/python3.9/site-packages/mujoco_py/cymj.o /venv/llm-bc/lib/python3.9/site-packages/mujoco_py/generated/_pyxbld_2.1.2.14_39_linuxcpuextensionbuilder/temp.linux-x86_64-cpython-39/venv/llm-bc/lib/python3.9/site-packages/mujoco_py/gl/osmesashim.o -L/root/.mujoco/mujoco210/bin -Wl,--enable-new-dtags,-rpath,/root/.mujoco/mujoco210/bin -lmujoco210 -lglewosmesa -lOSMesa -lGL -o /venv/llm-bc/lib/python3.9/site-packages/mujoco_py/generated/_pyxbld_2.1.2.14_39_linuxcpuextensionbuilder/lib.linux-x86_64-cpython-39/mujoco_py/cymj.cpython-39-x86_64-linux-gnu.so -fopenmp
+[2026-04-25 03:53:38,977][absl][INFO] - MUJOCO_GL=osmesa, attempting to import specified OpenGL backend.
+[2026-04-25 03:53:38,987][absl][INFO] - MuJoCo library version is: 2.3.7

2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/wandb/run-20260425_035339-ivkyq15y/files/output.log

@@ -0,0 +1,3 @@
+Eval MetaworldLowdimRunner 1/6:   0%|                                                                                                                               | 0/30 [00:00<?, ?it/s]/workspace/LLM-BC/llmbc/common/llfbench_util.py:39: UserWarning: Creating a tensor from a list of numpy.ndarrays is extremely slow. Please consider converting the list to a single numpy.ndarray with numpy.array() before converting to a tensor. (Triggered internally at ../torch/csrc/utils/tensor_new.cpp:275.)
+  obs = torch.tensor(obs, dtype=torch.float32).unsqueeze(dim=0).to(device)
+Training epoch 810:   0%|                                                                                            | 0/840 [00:00<?, ?it/s]                                                           

2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/wandb/run-20260425_035339-ivkyq15y/files/requirements.txt

@@ -0,0 +1,864 @@
+aiohappyeyeballs==2.6.1
+antlr4-python3-runtime==4.9.3
+asciitree==0.3.3
+async-timeout==5.0.1
+attrs==25.4.0
+certifi==2025.10.5
+charset-normalizer==3.4.4
+colorama==0.4.6
+dill==0.3.8
+docstring-parser==0.16
+eval_type_backport==0.2.2
+fasteners==0.20
+filelock==3.17.0
+frozenlist==1.8.0
+fsspec==2024.3.1
+gmpy2==2.2.1
+hf-xet==1.1.8
+idna==3.11
+llvmlite==0.39.1
+MarkupSafe==3.0.2
+mdurl==0.1.2
+mkl-service==2.4.0
+mpmath==1.3.0
+msgpack==1.1.1
+networkx==3.2.1
+packaging==25.0
+propcache==0.3.1
+psutil==7.0.0
+pycparser==2.23
+Pygments==2.19.1
+PyOpenGL==3.1.10
+PySocks==1.7.1
+tzdata==2025.3
+xxhash==3.5.0
+pytz==2025.2
+PyYAML==6.0.2
+regex==2025.9.1
+safetensors==0.5.3
+setuptools==80.9.0
+shtab==1.7.2
+six==1.17.0
+tqdm==4.67.1
+tensorboard==2.15.2
+uvloop==0.22.1
+wheel==0.45.1
+zipp==3.21.0
+tensorflow-estimator==2.15.0
+tensorboard-data-server==0.7.2
+google-auth-oauthlib==1.3.1
+tensorflow==2.15.1
+Jinja2==3.1.6
+markdown-it-py==2.2.0
+multiprocess==0.70.15
+numpy==1.23.5
+pillow==11.3.0
+pip==23.3.2
+python-dateutil==2.9.0.post0
+aiosignal==1.4.0
+brotlicffi==1.0.9.2
+hydra-core==1.2.0
+multidict==6.7.0
+pycares==4.10.0
+rich==14.2.0
+sympy==1.14.0
+typeguard==4.4.4
+aiodns==3.5.0
+urllib3==2.5.0
+yarl==1.18.0
+aiohttp==3.12.15
+pyarrow==21.0.0
+pyarrow-hotfix==0.7
+tokenizers==0.21.0
+av==10.0.0
+einops==0.4.1
+Bottleneck==1.4.2
+mkl_fft==1.3.11
+mkl_random==1.2.8
+numba==0.56.4
+numcodecs==0.12.1
+numexpr==2.10.1
+pandas==2.3.3
+zarr==2.12.0
+datasets==2.19.0
+transformers==4.47.1
+accelerate==1.0.1
+peft==0.14.0
+trl==0.11.4
+torch==2.2.2
+tabulate==0.9.0
+sentencepiece==0.2.0
+pyperclip==1.11.0
+PyOpenGL-accelerate==3.1.10
+pydub==0.25.1
+py-cpuinfo==9.0.0
+pure_eval==0.2.3
+ptyprocess==0.7.0
+nvidia-ml-py==13.595.45
+hjson==3.1.0
+gym-notices==0.1.0
+glfw==2.0.0
+Farama-Notifications==0.0.4
+websockets==15.0.1
+wcwidth==0.6.0
+uc-micro-py==1.0.3
+typing-inspection==0.4.2
+triton==2.2.0
+trimesh==4.11.5
+transforms3d==0.4.2
+traitlets==5.14.3
+threadpoolctl==3.6.0
+tenacity==9.1.2
+sniffio==1.3.1
+smmap==5.0.3
+setproctitle==1.3.7
+sentry-sdk==2.57.0
+semantic-version==2.10.0
+scipy==1.13.1
+rpds-py==0.27.1
+python-multipart==0.0.20
+pyparsing==3.3.2
+pynvml==13.0.1
+pydantic_core==2.41.5
+pycryptodomex==3.23.0
+pyasn1==0.6.3
+grpcio==1.80.0
+platformdirs==4.4.0
+pexpect==4.9.0
+parso==0.8.6
+orjson==3.11.5
+opencv-python==4.11.0.86
+stack-data==0.6.3
+omegaconf==2.2.1
+nvidia-nvtx-cu12==12.1.105
+nvidia-nvjitlink-cu12==12.9.86
+nvidia-nccl-cu12==2.19.3
+nvidia-curand-cu12==10.3.2.106
+nvidia-cufft-cu12==11.0.2.54
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cublas-cu12==12.1.3.1
+ninja==1.13.0
+narwhals==2.19.0
+lxml==6.0.3
+kiwisolver==1.4.7
+joblib==1.5.3
+jiter==0.13.0
+imageio-ffmpeg==0.6.0
+ImageIO==2.37.2
+h5py==3.14.0
+h11==0.16.0
+fonttools==4.60.2
+ffmpy==1.0.0
+fast_kinematics==0.2.2
+executing==2.2.1
+exceptiongroup==1.3.1
+docker-pycreds==0.4.0
+distro==1.9.0
+decorator==5.2.1
+dacite==1.9.2
+cycler==0.12.1
+contourpy==1.3.0
+cloudpickle==3.1.2
+click==8.1.8
+asttokens==3.0.1
+annotated-types==0.7.0
+annotated-doc==0.0.4
+aiofiles==25.1.0
+absl-py==2.3.1
+uvicorn==0.39.0
+tiktoken==0.8.0
+scikit-learn==1.6.1
+sapien==3.0.0b1
+referencing==0.36.2
+pydantic==2.12.5
+pyasn1_modules==0.4.2
+prompt_toolkit==3.0.52
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-cudnn-cu12==8.9.2.26
+nltk==3.9.2
+mujoco==2.3.7
+mdit-py-plugins==0.3.3
+matplotlib-inline==0.2.1
+matplotlib==3.7.5
+linkify-it-py==2.0.3
+jedi==0.19.2
+huggingface-hub==0.26.2
+httpcore==1.0.9
+gymnasium==0.29.1
+google-pasta==0.2.0
+gitdb==4.0.12
+cryptography==46.0.7
+blobfile==3.0.0
+anyio==4.12.1
+tyro==0.9.1
+toppra==0.6.6
+starlette==0.49.3
+seaborn==0.13.2
+rouge_score==0.1.2
+pynndescent==0.6.0
+nvidia-cusolver-cu12==11.4.5.107
+jsonschema-specifications==2025.9.1
+ipython==8.18.1
+httpx==0.28.1
+gymnasium-robotics==1.2.0
+google-auth==2.49.2
+GitPython==3.1.46
+diffusers==0.31.0
+wandb==0.18.6
+umap-learn==0.5.9.post2
+requests-oauthlib==2.0.0
+openai==2.8.1
+mplib==0.1.1
+jsonschema==4.25.1
+gradio_client==0.2.9
+google-genai==1.47.0
+fastapi==0.128.8
+torchvision==0.17.2
+torchaudio==2.2.2
+sentence-transformers==3.2.1
+pytorch-seed==0.2.0
+deepspeed==0.16.1
+bitsandbytes==0.45.0
+altair==6.0.0
+timm==1.0.22
+gradio==3.36.1
+evaluate==0.4.3
+arm_pytorch_utilities==0.5.0
+tensorflow-io-gcs-filesystem==0.37.1
+Markdown==3.9
+pybullet==3.2.6
+wrapt==1.14.2
+tensorflow-probability==0.23.0
+parse==1.19.1
+dm-tree==0.1.8
+termcolor==3.1.0
+requests==2.32.0
+cffi==2.0.0
+opt_einsum==3.4.0
+labmaze==1.0.6
+typing_extensions==4.15.0
+importlib-resources==5.13.0
+importlib-metadata==5.2.0
+FLAML==2.3.6
+dm-env==1.6
+diskcache==5.6.3
+Cython==0.29.37
+mujoco-py==2.1.2.14
+jaxlib==0.4.30
+dm-control==1.0.14
+cmudict==1.0.13
+syllables==1.0.9
+metaworld==2.0.0
+jax==0.4.30
+highway-env==1.9.1
+gym_bandits==0.0.2
+d4rl==1.1
+pyautogen==0.1.0
+protobuf==4.25.9
+tf-agents==0.19.0
+oauthlib==3.3.1
+pytorch-kinematics==0.7.5
+pygame==2.6.1
+gin-config==0.5.0
+mani_skill==3.0.0b20
+tifffile==2024.8.30
+Shapely==1.8.4
+PyWavelets==1.6.0
+Werkzeug==3.1.8
+astunparse==1.6.3
+scikit-image==0.19.3
+pymunk==6.2.1
+ml-dtypes==0.3.2
+gast==0.7.0
+stable-baselines3==2.2.1
+keras==2.15.0
+libclang==18.1.1
+flatbuffers==25.12.19
+gym==0.23.1
+llmbc==0.0.0
+mjrl==1.0.0
+llfbench==0.1.0
+aiohappyeyeballs==2.6.1
+antlr4-python3-runtime==4.9.3
+asciitree==0.3.3
+async-timeout==5.0.1
+attrs==25.4.0
+certifi==2025.10.5
+charset-normalizer==3.4.4
+colorama==0.4.6
+dill==0.3.8
+docstring-parser==0.16
+eval_type_backport==0.2.2
+fasteners==0.20
+filelock==3.17.0
+frozenlist==1.8.0
+fsspec==2024.3.1
+gmpy2==2.2.1
+hf-xet==1.1.8
+idna==3.11
+llvmlite==0.39.1
+MarkupSafe==3.0.2
+mdurl==0.1.2
+mkl-service==2.4.0
+mpmath==1.3.0
+msgpack==1.1.1
+networkx==3.2.1
+packaging==25.0
+propcache==0.3.1
+psutil==7.0.0
+pycparser==2.23
+Pygments==2.19.1
+PyOpenGL==3.1.10
+PySocks==1.7.1
+tzdata==2025.3
+xxhash==3.5.0
+pytz==2025.2
+PyYAML==6.0.2
+regex==2025.9.1
+safetensors==0.5.3
+setuptools==80.9.0
+shtab==1.7.2
+six==1.17.0
+tqdm==4.67.1
+tensorboard==2.15.2
+uvloop==0.22.1
+wheel==0.45.1
+zipp==3.21.0
+tensorflow-estimator==2.15.0
+tensorboard-data-server==0.7.2
+google-auth-oauthlib==1.3.1
+tensorflow==2.15.1
+Jinja2==3.1.6
+markdown-it-py==2.2.0
+multiprocess==0.70.15
+numpy==1.23.5
+pillow==11.3.0
+pip==23.3.2
+python-dateutil==2.9.0.post0
+aiosignal==1.4.0
+brotlicffi==1.0.9.2
+hydra-core==1.2.0
+multidict==6.7.0
+pycares==4.10.0
+rich==14.2.0
+sympy==1.14.0
+typeguard==4.4.4
+aiodns==3.5.0
+urllib3==2.5.0
+yarl==1.18.0
+aiohttp==3.12.15
+pyarrow==21.0.0
+pyarrow-hotfix==0.7
+tokenizers==0.21.0
+av==10.0.0
+einops==0.4.1
+Bottleneck==1.4.2
+mkl_fft==1.3.11
+mkl_random==1.2.8
+numba==0.56.4
+numcodecs==0.12.1
+numexpr==2.10.1
+pandas==2.3.3
+zarr==2.12.0
+datasets==2.19.0
+transformers==4.47.1
+accelerate==1.0.1
+peft==0.14.0
+trl==0.11.4
+torch==2.2.2
+tabulate==0.9.0
+sentencepiece==0.2.0
+pyperclip==1.11.0
+PyOpenGL-accelerate==3.1.10
+pydub==0.25.1
+py-cpuinfo==9.0.0
+pure_eval==0.2.3
+ptyprocess==0.7.0
+nvidia-ml-py==13.595.45
+hjson==3.1.0
+gym-notices==0.1.0
+glfw==2.0.0
+Farama-Notifications==0.0.4
+websockets==15.0.1
+wcwidth==0.6.0
+uc-micro-py==1.0.3
+typing-inspection==0.4.2
+triton==2.2.0
+trimesh==4.11.5
+transforms3d==0.4.2
+traitlets==5.14.3
+threadpoolctl==3.6.0
+tenacity==9.1.2
+sniffio==1.3.1
+smmap==5.0.3
+setproctitle==1.3.7
+sentry-sdk==2.57.0
+semantic-version==2.10.0
+scipy==1.13.1
+rpds-py==0.27.1
+python-multipart==0.0.20
+pyparsing==3.3.2
+pynvml==13.0.1
+pydantic_core==2.41.5
+pycryptodomex==3.23.0
+pyasn1==0.6.3
+grpcio==1.80.0
+platformdirs==4.4.0
+pexpect==4.9.0
+parso==0.8.6
+orjson==3.11.5
+opencv-python==4.11.0.86
+stack-data==0.6.3
+omegaconf==2.2.1
+nvidia-nvtx-cu12==12.1.105
+nvidia-nvjitlink-cu12==12.9.86
+nvidia-nccl-cu12==2.19.3
+nvidia-curand-cu12==10.3.2.106
+nvidia-cufft-cu12==11.0.2.54
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cublas-cu12==12.1.3.1
+ninja==1.13.0
+narwhals==2.19.0
+lxml==6.0.3
+kiwisolver==1.4.7
+joblib==1.5.3
+jiter==0.13.0
+imageio-ffmpeg==0.6.0
+ImageIO==2.37.2
+h5py==3.14.0
+h11==0.16.0
+fonttools==4.60.2
+ffmpy==1.0.0
+fast_kinematics==0.2.2
+executing==2.2.1
+exceptiongroup==1.3.1
+docker-pycreds==0.4.0
+distro==1.9.0
+decorator==5.2.1
+dacite==1.9.2
+cycler==0.12.1
+contourpy==1.3.0
+cloudpickle==3.1.2
+click==8.1.8
+asttokens==3.0.1
+annotated-types==0.7.0
+annotated-doc==0.0.4
+aiofiles==25.1.0
+absl-py==2.3.1
+uvicorn==0.39.0
+tiktoken==0.8.0
+scikit-learn==1.6.1
+sapien==3.0.0b1
+referencing==0.36.2
+pydantic==2.12.5
+pyasn1_modules==0.4.2
+prompt_toolkit==3.0.52
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-cudnn-cu12==8.9.2.26
+nltk==3.9.2
+mujoco==2.3.7
+mdit-py-plugins==0.3.3
+matplotlib-inline==0.2.1
+matplotlib==3.7.5
+linkify-it-py==2.0.3
+jedi==0.19.2
+huggingface-hub==0.26.2
+httpcore==1.0.9
+gymnasium==0.29.1
+google-pasta==0.2.0
+gitdb==4.0.12
+cryptography==46.0.7
+blobfile==3.0.0
+anyio==4.12.1
+tyro==0.9.1
+toppra==0.6.6
+starlette==0.49.3
+seaborn==0.13.2
+rouge_score==0.1.2
+pynndescent==0.6.0
+nvidia-cusolver-cu12==11.4.5.107
+jsonschema-specifications==2025.9.1
+ipython==8.18.1
+httpx==0.28.1
+gymnasium-robotics==1.2.0
+google-auth==2.49.2
+GitPython==3.1.46
+diffusers==0.31.0
+wandb==0.18.6
+umap-learn==0.5.9.post2
+requests-oauthlib==2.0.0
+openai==2.8.1
+mplib==0.1.1
+jsonschema==4.25.1
+gradio_client==0.2.9
+google-genai==1.47.0
+fastapi==0.128.8
+torchvision==0.17.2
+torchaudio==2.2.2
+sentence-transformers==3.2.1
+pytorch-seed==0.2.0
+deepspeed==0.16.1
+bitsandbytes==0.45.0
+altair==6.0.0
+timm==1.0.22
+gradio==3.36.1
+evaluate==0.4.3
+arm_pytorch_utilities==0.5.0
+tensorflow-io-gcs-filesystem==0.37.1
+Markdown==3.9
+pybullet==3.2.6
+wrapt==1.14.2
+tensorflow-probability==0.23.0
+parse==1.19.1
+dm-tree==0.1.8
+termcolor==3.1.0
+requests==2.32.0
+cffi==2.0.0
+opt_einsum==3.4.0
+labmaze==1.0.6
+typing_extensions==4.15.0
+importlib-resources==5.13.0
+importlib-metadata==5.2.0
+FLAML==2.3.6
+dm-env==1.6
+diskcache==5.6.3
+Cython==0.29.37
+mujoco-py==2.1.2.14
+jaxlib==0.4.30
+dm-control==1.0.14
+cmudict==1.0.13
+syllables==1.0.9
+metaworld==2.0.0
+jax==0.4.30
+highway-env==1.9.1
+gym_bandits==0.0.2
+d4rl==1.1
+pyautogen==0.1.0
+protobuf==4.25.9
+tf-agents==0.19.0
+oauthlib==3.3.1
+pytorch-kinematics==0.7.5
+pygame==2.6.1
+gin-config==0.5.0
+mani_skill==3.0.0b20
+tifffile==2024.8.30
+Shapely==1.8.4
+PyWavelets==1.6.0
+Werkzeug==3.1.8
+astunparse==1.6.3
+scikit-image==0.19.3
+pymunk==6.2.1
+ml-dtypes==0.3.2
+gast==0.7.0
+stable-baselines3==2.2.1
+keras==2.15.0
+libclang==18.1.1
+flatbuffers==25.12.19
+gym==0.23.1
+llmbc==0.0.0
+mjrl==1.0.0
+llfbench==0.1.0
+llmbc==0.0.0
+llmbc==0.0.0
+aiohappyeyeballs==2.6.1
+antlr4-python3-runtime==4.9.3
+asciitree==0.3.3
+async-timeout==5.0.1
+attrs==25.4.0
+certifi==2025.10.5
+charset-normalizer==3.4.4
+colorama==0.4.6
+dill==0.3.8
+docstring-parser==0.16
+eval_type_backport==0.2.2
+fasteners==0.20
+filelock==3.17.0
+frozenlist==1.8.0
+fsspec==2024.3.1
+gmpy2==2.2.1
+hf-xet==1.1.8
+idna==3.11
+llvmlite==0.39.1
+MarkupSafe==3.0.2
+mdurl==0.1.2
+mkl-service==2.4.0
+mpmath==1.3.0
+msgpack==1.1.1
+networkx==3.2.1
+packaging==25.0
+propcache==0.3.1
+psutil==7.0.0
+pycparser==2.23
+Pygments==2.19.1
+PyOpenGL==3.1.10
+PySocks==1.7.1
+tzdata==2025.3
+xxhash==3.5.0
+pytz==2025.2
+PyYAML==6.0.2
+regex==2025.9.1
+safetensors==0.5.3
+setuptools==80.9.0
+shtab==1.7.2
+six==1.17.0
+tqdm==4.67.1
+tensorboard==2.15.2
+uvloop==0.22.1
+wheel==0.45.1
+zipp==3.21.0
+tensorflow-estimator==2.15.0
+tensorboard-data-server==0.7.2
+google-auth-oauthlib==1.3.1
+tensorflow==2.15.1
+Jinja2==3.1.6
+markdown-it-py==2.2.0
+multiprocess==0.70.15
+numpy==1.23.5
+pillow==11.3.0
+pip==23.3.2
+python-dateutil==2.9.0.post0
+aiosignal==1.4.0
+brotlicffi==1.0.9.2
+hydra-core==1.2.0
+multidict==6.7.0
+pycares==4.10.0
+rich==14.2.0
+sympy==1.14.0
+typeguard==4.4.4
+aiodns==3.5.0
+urllib3==2.5.0
+yarl==1.18.0
+aiohttp==3.12.15
+pyarrow==21.0.0
+pyarrow-hotfix==0.7
+tokenizers==0.21.0
+av==10.0.0
+einops==0.4.1
+Bottleneck==1.4.2
+mkl_fft==1.3.11
+mkl_random==1.2.8
+numba==0.56.4
+numcodecs==0.12.1
+numexpr==2.10.1
+pandas==2.3.3
+zarr==2.12.0
+datasets==2.19.0
+transformers==4.47.1
+accelerate==1.0.1
+peft==0.14.0
+trl==0.11.4
+torch==2.2.2
+tabulate==0.9.0
+sentencepiece==0.2.0
+pyperclip==1.11.0
+PyOpenGL-accelerate==3.1.10
+pydub==0.25.1
+py-cpuinfo==9.0.0
+pure_eval==0.2.3
+ptyprocess==0.7.0
+nvidia-ml-py==13.595.45
+hjson==3.1.0
+gym-notices==0.1.0
+glfw==2.0.0
+Farama-Notifications==0.0.4
+websockets==15.0.1
+wcwidth==0.6.0
+uc-micro-py==1.0.3
+typing-inspection==0.4.2
+triton==2.2.0
+trimesh==4.11.5
+transforms3d==0.4.2
+traitlets==5.14.3
+threadpoolctl==3.6.0
+tenacity==9.1.2
+sniffio==1.3.1
+smmap==5.0.3
+setproctitle==1.3.7
+sentry-sdk==2.57.0
+semantic-version==2.10.0
+scipy==1.13.1
+rpds-py==0.27.1
+python-multipart==0.0.20
+pyparsing==3.3.2
+pynvml==13.0.1
+pydantic_core==2.41.5
+pycryptodomex==3.23.0
+pyasn1==0.6.3
+grpcio==1.80.0
+platformdirs==4.4.0
+pexpect==4.9.0
+parso==0.8.6
+orjson==3.11.5
+opencv-python==4.11.0.86
+stack-data==0.6.3
+omegaconf==2.2.1
+nvidia-nvtx-cu12==12.1.105
+nvidia-nvjitlink-cu12==12.9.86
+nvidia-nccl-cu12==2.19.3
+nvidia-curand-cu12==10.3.2.106
+nvidia-cufft-cu12==11.0.2.54
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cublas-cu12==12.1.3.1
+ninja==1.13.0
+narwhals==2.19.0
+lxml==6.0.3
+kiwisolver==1.4.7
+joblib==1.5.3
+jiter==0.13.0
+imageio-ffmpeg==0.6.0
+ImageIO==2.37.2
+h5py==3.14.0
+h11==0.16.0
+fonttools==4.60.2
+ffmpy==1.0.0
+fast_kinematics==0.2.2
+executing==2.2.1
+exceptiongroup==1.3.1
+docker-pycreds==0.4.0
+distro==1.9.0
+decorator==5.2.1
+dacite==1.9.2
+cycler==0.12.1
+contourpy==1.3.0
+cloudpickle==3.1.2
+click==8.1.8
+asttokens==3.0.1
+annotated-types==0.7.0
+annotated-doc==0.0.4
+aiofiles==25.1.0
+absl-py==2.3.1
+uvicorn==0.39.0
+tiktoken==0.8.0
+scikit-learn==1.6.1
+sapien==3.0.0b1
+referencing==0.36.2
+pydantic==2.12.5
+pyasn1_modules==0.4.2
+prompt_toolkit==3.0.52
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-cudnn-cu12==8.9.2.26
+nltk==3.9.2
+mujoco==2.3.7
+mdit-py-plugins==0.3.3
+matplotlib-inline==0.2.1
+matplotlib==3.7.5
+linkify-it-py==2.0.3
+jedi==0.19.2
+huggingface-hub==0.26.2
+httpcore==1.0.9
+gymnasium==0.29.1
+google-pasta==0.2.0
+gitdb==4.0.12
+cryptography==46.0.7
+blobfile==3.0.0
+anyio==4.12.1
+tyro==0.9.1
+toppra==0.6.6
+starlette==0.49.3
+seaborn==0.13.2
+rouge_score==0.1.2
+pynndescent==0.6.0
+nvidia-cusolver-cu12==11.4.5.107
+jsonschema-specifications==2025.9.1
+ipython==8.18.1
+httpx==0.28.1
+gymnasium-robotics==1.2.0
+google-auth==2.49.2
+GitPython==3.1.46
+diffusers==0.31.0
+wandb==0.18.6
+umap-learn==0.5.9.post2
+requests-oauthlib==2.0.0
+openai==2.8.1
+mplib==0.1.1
+jsonschema==4.25.1
+gradio_client==0.2.9
+google-genai==1.47.0
+fastapi==0.128.8
+torchvision==0.17.2
+torchaudio==2.2.2
+sentence-transformers==3.2.1
+pytorch-seed==0.2.0
+deepspeed==0.16.1
+bitsandbytes==0.45.0
+altair==6.0.0
+timm==1.0.22
+gradio==3.36.1
+evaluate==0.4.3
+arm_pytorch_utilities==0.5.0
+tensorflow-io-gcs-filesystem==0.37.1
+Markdown==3.9
+pybullet==3.2.6
+wrapt==1.14.2
+tensorflow-probability==0.23.0
+parse==1.19.1
+dm-tree==0.1.8
+termcolor==3.1.0
+requests==2.32.0
+cffi==2.0.0
+opt_einsum==3.4.0
+labmaze==1.0.6
+typing_extensions==4.15.0
+importlib-resources==5.13.0
+importlib-metadata==5.2.0
+FLAML==2.3.6
+dm-env==1.6
+diskcache==5.6.3
+Cython==0.29.37
+mujoco-py==2.1.2.14
+jaxlib==0.4.30
+dm-control==1.0.14
+cmudict==1.0.13
+syllables==1.0.9
+metaworld==2.0.0
+jax==0.4.30
+highway-env==1.9.1
+gym_bandits==0.0.2
+d4rl==1.1
+pyautogen==0.1.0
+protobuf==4.25.9
+tf-agents==0.19.0
+oauthlib==3.3.1
+pytorch-kinematics==0.7.5
+pygame==2.6.1
+gin-config==0.5.0
+mani_skill==3.0.0b20
+tifffile==2024.8.30
+Shapely==1.8.4
+PyWavelets==1.6.0
+Werkzeug==3.1.8
+astunparse==1.6.3
+scikit-image==0.19.3
+pymunk==6.2.1
+ml-dtypes==0.3.2
+gast==0.7.0
+stable-baselines3==2.2.1
+keras==2.15.0
+libclang==18.1.1
+flatbuffers==25.12.19
+gym==0.23.1
+llmbc==0.0.0
+mjrl==1.0.0
+llfbench==0.1.0
+autocommand==2.2.2
+backports.tarfile==1.2.0
+importlib_metadata==8.0.0
+inflect==7.3.1
+jaraco.collections==5.1.0
+jaraco.context==5.3.0
+jaraco.functools==4.0.1
+jaraco.text==3.12.1
+more-itertools==10.3.0
+packaging==24.2
+platformdirs==4.2.2
+tomli==2.0.1
+typeguard==4.3.0
+typing_extensions==4.12.2
+wheel==0.45.1
+zipp==3.19.2

2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/wandb/run-20260425_035339-ivkyq15y/files/wandb-metadata.json

@@ -0,0 +1,56 @@
+{
+  "os": "Linux-5.15.0-168-generic-x86_64-with-glibc2.39",
+  "python": "3.9.25",
+  "startedAt": "2026-04-25T03:53:39.984340Z",
+  "args": [
+    "--config-path",
+    "config/mix_data",
+    "--config-name",
+    "llmdp_sweep-v2_mix(csp3).yaml",
+    "training.seed=42"
+  ],
+  "program": "/workspace/LLM-BC/./train.py",
+  "codePath": "train.py",
+  "git": {
+    "remote": "https://github.com/CHYang25/LLM-BC.git",
+    "commit": "0131b06a1134730d8afeb66e613e0eb72963e3c4"
+  },
+  "email": "chris920325@gmail.com",
+  "root": "/workspace/LLM-BC/data/outputs/2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2",
+  "host": "c8ef52649900",
+  "username": "root",
+  "executable": "/venv/llm-bc/bin/python3",
+  "codePathLocal": "train.py",
+  "cpu_count": 72,
+  "cpu_count_logical": 144,
+  "gpu": "NVIDIA GeForce RTX 4090",
+  "gpu_count": 2,
+  "disk": {
+    "/": {
+      "total": "545460846592",
+      "used": "476012007424"
+    }
+  },
+  "memory": {
+    "total": "810942898176"
+  },
+  "cpu": {
+    "count": 72,
+    "countLogical": 144
+  },
+  "gpu_nvidia": [
+    {
+      "name": "NVIDIA GeForce RTX 4090",
+      "memoryTotal": "25757220864",
+      "cudaCores": 16384,
+      "architecture": "Ada"
+    },
+    {
+      "name": "NVIDIA GeForce RTX 4090",
+      "memoryTotal": "25757220864",
+      "cudaCores": 16384,
+      "architecture": "Ada"
+    }
+  ],
+  "cudaVersion": "12.8"
+}

2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/wandb/run-20260425_035339-ivkyq15y/run-ivkyq15y.wandb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6f62a38a5e8914a6b4acd1225f025ff4178cacf5c7edad38ce56e1f4ae5f08ff
+size 679772160

2026.04.25/03.52.02_train_llm_diffusion_unet_lowdim_sweep-v2/wandb/wandb-resume.json

@@ -0,0 +1 @@
+{"run_id": "ivkyq15y"}

2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/.hydra/config.yaml

@@ -0,0 +1,193 @@
+name: train_llm_diffusion_unet_lowdim
+_target_: llmbc.workspace.train_llm_diffusion_unet_lowdim_workspace.TrainLLMDiffusionUnetLowdimWorkspace
+obs_dim: ${task.obs_dim}
+action_dim: ${task.action_dim}
+task_name: ${task.name}
+exp_name: default
+model_name: ${llm.name}
+horizon: 1
+n_obs_steps: 1
+n_action_steps: 1
+n_latency_steps: 0
+past_action_visible: false
+keypoint_visible_rate: 1.0
+obs_as_local_cond: false
+obs_as_global_cond: true
+pred_action_steps_only: false
+llm_orig_expert_feedback: true
+llm_do_sample: false
+policy:
+  _target_: llmbc.policy.llm_diffusion_unet_lowdim_policy.LLMDiffusionUnetLowdimPolicy
+  model:
+    _target_: llmbc.model.diffusion.conditional_unet1d.ConditionalUnet1D
+    input_dim: '${eval: ${task.action_dim} if ${obs_as_local_cond} or ${obs_as_global_cond}
+      else ${task.obs_dim} + ${task.action_dim}}'
+    local_cond_dim: '${eval: ${task.obs_dim} if ${obs_as_local_cond} else None}'
+    global_cond_dim: '${eval: ${task.obs_dim}*${n_obs_steps} if ${obs_as_global_cond}
+      else None}'
+    diffusion_step_embed_dim: 256
+    down_dims:
+    - 256
+    - 512
+    - 1024
+    kernel_size: 5
+    n_groups: 8
+    cond_predict_scale: true
+  noise_scheduler:
+    _target_: diffusers.schedulers.scheduling_ddpm.DDPMScheduler
+    num_train_timesteps: 100
+    beta_start: 0.0001
+    beta_end: 0.02
+    beta_schedule: squaredcos_cap_v2
+    variance_type: fixed_small
+    clip_sample: true
+    prediction_type: epsilon
+  horizon: ${horizon}
+  obs_dim: ${obs_dim}
+  action_dim: ${action_dim}
+  n_action_steps: ${eval:'${n_action_steps}+${n_latency_steps}'}
+  n_obs_steps: ${n_obs_steps}
+  num_inference_steps: 100
+  obs_as_local_cond: ${obs_as_local_cond}
+  obs_as_global_cond: ${obs_as_global_cond}
+  pred_action_steps_only: ${pred_action_steps_only}
+  oa_step_convention: true
+  llm_discriminator:
+    _target_: llmbc.discriminator.llm_ce_discriminator.LLMCEDiscriminator
+    task_id: ${task_name}
+    llm_translator:
+      _target_: llmbc.translator.llm_translator.LLMTranslator
+      cfg: ${llm}
+      obs_dim: ${obs_dim}
+      action_dim: ${action_dim}
+      horizon: ${horizon}
+      n_obs_steps: ${n_obs_steps}
+      n_action_steps: ${eval:'${n_action_steps}+${n_latency_steps}'}
+  loss_dp_weight: 1.0
+  loss_llm_weight: 0.001
+  normalize_llm_loss: true
+  reweight_llm_loss: true
+ema:
+  _target_: llmbc.model.diffusion.ema_model.EMAModel
+  update_after_step: 0
+  inv_gamma: 1.0
+  power: 0.75
+  min_value: 0.0
+  max_value: 0.9999
+dataloader:
+  batch_size: 16
+  num_workers: 0
+  shuffle: true
+  pin_memory: false
+  persistent_workers: false
+val_dataloader:
+  batch_size: 16
+  num_workers: 0
+  shuffle: false
+  pin_memory: false
+  persistent_workers: false
+optimizer:
+  _target_: torch.optim.AdamW
+  lr: 0.0001
+  betas:
+  - 0.95
+  - 0.999
+  eps: 1.0e-08
+  weight_decay: 1.0e-06
+training:
+  device: cuda:0
+  seed: 43
+  debug: false
+  resume: true
+  lr_scheduler: cosine
+  lr_warmup_steps: 500
+  num_epochs: 1001
+  gradient_accumulate_every: 16
+  use_ema: true
+  rollout_every: 5
+  checkpoint_every: 5
+  val_every: 1
+  sample_every: 5
+  max_train_steps: null
+  max_val_steps: null
+  tqdm_interval_sec: 1.0
+logging:
+  project: ${task.name}-training-mix
+  resume: true
+  mode: online
+  name: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
+  tags:
+  - ${name}
+  - ${task_name}
+  - ${exp_name}
+  id: null
+  group: null
+checkpoint:
+  topk:
+    monitor_key: test_success_rate
+    mode: max
+    k: 5
+    format_str: epoch={epoch:04d}-test_success_rate={test_success_rate:.3f}.ckpt
+  save_last_ckpt: true
+  save_last_snapshot: false
+multi_run:
+  run_dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+  wandb_name_base: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
+task:
+  name: sweep-v2
+  obs_dim: 9
+  action_dim: 4
+  env_runner:
+    _target_: llmbc.env_runner.metaworld_lowdim_runner.MetaworldLowdimRunner
+    env_name: llf-metaworld-sweep-v2
+    n_train: 10
+    n_test: 50
+    n_envs: 10
+    max_steps: 30
+    n_obs_steps: ${n_obs_steps}
+    n_action_steps: ${n_action_steps}
+    instruction_type: b
+    feedback_type:
+    - hp
+    - hn
+    - fp
+    visual: false
+    discount: 0.9
+  dataset:
+    _target_: llmbc.dataset.metaworld_lowdim_dataset.MetaworldLowdimDataset
+    data_path: datasets/sweep-v2-mix(csp3).pt
+    data_path2: datasets/sweep-v2.pt
+    horizon: ${horizon}
+    pad_before: ${eval:'${n_obs_steps}-1'}
+    pad_after: ${eval:'${n_action_steps}-1'}
+    obs_eef_target: true
+    use_manual_normalizer: false
+    val_ratio: 0.1
+    dummy_normalizer: false
+  instructor:
+    _target_: llmbc.translator.instructor.metaworld_instructor.sweep_v2_instructor.SweepV2Instructor
+llm:
+  name: HuggingFaceTB/SmolLM2-135M-Instruct
+  model_name: SmolLM2-135M-Instruct
+  config_target: llmbc.model.llm.llama_lowdim_model.LowdimLlamaConfig
+  causal_lm_target: llmbc.model.llm.llama_lowdim_model.LowdimLlamaForCausalLM
+  use_quantization: false
+  use_joint_mlp_projector: true
+  llm_mode: ete-finetuned
+  finetune_mode: orig
+  checkpoint: data/outputs/2025.11.01/23.03.01_train_llm_lowdim_sweep-v2/HuggingFaceTB/SmolLM2-135M-Instruct-finetuned-sweep-v2/checkpoint-5690
+  max_length: 100
+  lora_config:
+    r: 32
+    lora_alpha: 64
+    lora_dropout: 0.05
+    bias: none
+    task_type: CAUSAL_LM
+  prompter:
+    _target_: llmbc.translator.prompter.smollm2_prompter.SmolLM2Prompter
+    use_joint_mlp_projector: true
+  hydra:
+    job:
+      override_dirname: ${model_name}
+    run:
+      dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${model_name}

2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/.hydra/hydra.yaml

@@ -0,0 +1,155 @@
+hydra:
+  run:
+    dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+  sweep:
+    dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+    subdir: ${hydra.job.num}
+  launcher:
+    _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
+  sweeper:
+    _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
+    max_batch_size: null
+    params: null
+  help:
+    app_name: ${hydra.job.name}
+    header: '${hydra.help.app_name} is powered by Hydra.
+
+      '
+    footer: 'Powered by Hydra (https://hydra.cc)
+
+      Use --hydra-help to view Hydra specific help
+
+      '
+    template: '${hydra.help.header}
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (group=option)
+
+
+      $APP_CONFIG_GROUPS
+
+
+      == Config ==
+
+      Override anything in the config (foo.bar=value)
+
+
+      $CONFIG
+
+
+      ${hydra.help.footer}
+
+      '
+  hydra_help:
+    template: 'Hydra (${hydra.runtime.version})
+
+      See https://hydra.cc for more info.
+
+
+      == Flags ==
+
+      $FLAGS_HELP
+
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (For example, append hydra/job_logging=disabled
+      to command line)
+
+
+      $HYDRA_CONFIG_GROUPS
+
+
+      Use ''--cfg hydra'' to Show the Hydra config.
+
+      '
+    hydra_help: ???
+  hydra_logging:
+    version: 1
+    formatters:
+      simple:
+        format: '[%(asctime)s][HYDRA] %(message)s'
+    handlers:
+      console:
+        class: logging.StreamHandler
+        formatter: simple
+        stream: ext://sys.stdout
+    root:
+      level: INFO
+      handlers:
+      - console
+    loggers:
+      logging_example:
+        level: DEBUG
+    disable_existing_loggers: false
+  job_logging:
+    version: 1
+    formatters:
+      simple:
+        format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
+    handlers:
+      console:
+        class: logging.StreamHandler
+        formatter: simple
+        stream: ext://sys.stdout
+      file:
+        class: logging.FileHandler
+        formatter: simple
+        filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
+    root:
+      level: INFO
+      handlers:
+      - console
+      - file
+    disable_existing_loggers: false
+  env: {}
+  mode: RUN
+  searchpath: []
+  callbacks: {}
+  output_subdir: .hydra
+  overrides:
+    hydra:
+    - hydra.mode=RUN
+    task:
+    - training.seed=43
+  job:
+    name: train
+    chdir: null
+    override_dirname: training.seed=43
+    id: ???
+    num: ???
+    config_name: llmdp_sweep-v2_mix(csp3).yaml
+    env_set: {}
+    env_copy: []
+    config:
+      override_dirname:
+        kv_sep: '='
+        item_sep: ','
+        exclude_keys: []
+  runtime:
+    version: 1.2.0
+    version_base: '1.2'
+    cwd: /workspace/LLM-BC
+    config_sources:
+    - path: hydra.conf
+      schema: pkg
+      provider: hydra
+    - path: /workspace/LLM-BC/config/mix_data
+      schema: file
+      provider: main
+    - path: ''
+      schema: structured
+      provider: schema
+    output_dir: /workspace/LLM-BC/data/outputs/2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2
+    choices:
+      hydra/env: default
+      hydra/callbacks: null
+      hydra/job_logging: default
+      hydra/hydra_logging: default
+      hydra/hydra_help: default
+      hydra/help: default
+      hydra/sweeper: basic
+      hydra/launcher: basic
+      hydra/output: default
+  verbose: false

2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/.hydra/overrides.yaml

@@ -0,0 +1 @@
+- training.seed=43

2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/checkpoints/epoch=0080-test_success_rate=0.660.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6bfd442f2e02f331bdff7fb6121dbb8a90fd5005617e5cda218512d68da887d9
+size 1040498994

2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/checkpoints/epoch=0140-test_success_rate=0.660.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8f26933fbdbb921bba60c3bf67aeb9e563b89e9de228223858ba97c9e3d003da
+size 1040498994

2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/checkpoints/epoch=0145-test_success_rate=0.700.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ab4f89dd29bc729aada4b7dfbf0a924d3fdb2fb44229cbd8a9927aa0e3cc2e7a
+size 1040498994

2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/checkpoints/epoch=0245-test_success_rate=0.660.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:71760637103d97c7783a5d040411ab9576b239d3c5d15a3ea9207086c25683e4
+size 1040498994

2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/checkpoints/epoch=0280-test_success_rate=0.680.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:093abfad164dfd864e325cac05f32d0b5ece97bd0686b5de1dc0a312c3921ebe
+size 1040498994

2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/checkpoints/latest.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:77e3a2c110730ff3b501b8b39ff1868c5b8596dc948a4ecaaf0964db09ff5d18
+size 1040498994

2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/logs.json.txt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6ab45032bcd8aa92a64f70d6d35be1faf3edfb62df58a002e8c105b319c20e05
+size 121188663

2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/train.log

@@ -0,0 +1,14 @@
+[2026-04-25 03:52:13,526][numexpr.utils][INFO] - Note: detected 144 virtual cores but NumExpr set to maximum of 64, check "NUMEXPR_MAX_THREADS" environment variable.
+[2026-04-25 03:52:13,526][numexpr.utils][INFO] - Note: NumExpr detected 144 cores but "NUMEXPR_MAX_THREADS" not set, so enforcing safe limit of 16.
+[2026-04-25 03:52:13,526][numexpr.utils][INFO] - NumExpr defaulting to 16 threads.
+[2026-04-25 03:52:15,747][datasets][INFO] - PyTorch version 2.2.2 available.
+[2026-04-25 03:52:15,748][datasets][INFO] - TensorFlow version 2.15.1 available.
+[2026-04-25 03:52:15,749][datasets][INFO] - JAX version 0.4.30 available.
+[2026-04-25 03:52:17,443][llmbc.model.diffusion.conditional_unet1d][INFO] - number of parameters: 6.501479e+07
+[2026-04-25 03:52:56,097][root][INFO] - running build_ext
+[2026-04-25 03:52:56,100][root][INFO] - building 'mujoco_py.cymj' extension
+[2026-04-25 03:52:56,101][root][INFO] - gcc -pthread -B /venv/llm-bc/compiler_compat -Wno-unused-result -Wsign-compare -DNDEBUG -O2 -Wall -fPIC -O2 -isystem /venv/llm-bc/include -I/venv/llm-bc/include -fPIC -O2 -isystem /venv/llm-bc/include -fPIC -I/venv/llm-bc/lib/python3.9/site-packages/mujoco_py -I/root/.mujoco/mujoco210/include -I/venv/llm-bc/lib/python3.9/site-packages/numpy/core/include -I/venv/llm-bc/include/python3.9 -c /venv/llm-bc/lib/python3.9/site-packages/mujoco_py/cymj.c -o /venv/llm-bc/lib/python3.9/site-packages/mujoco_py/generated/_pyxbld_2.1.2.14_39_linuxcpuextensionbuilder/temp.linux-x86_64-cpython-39/venv/llm-bc/lib/python3.9/site-packages/mujoco_py/cymj.o -fopenmp -w
+[2026-04-25 03:53:37,867][root][INFO] - gcc -pthread -B /venv/llm-bc/compiler_compat -Wno-unused-result -Wsign-compare -DNDEBUG -O2 -Wall -fPIC -O2 -isystem /venv/llm-bc/include -I/venv/llm-bc/include -fPIC -O2 -isystem /venv/llm-bc/include -fPIC -I/venv/llm-bc/lib/python3.9/site-packages/mujoco_py -I/root/.mujoco/mujoco210/include -I/venv/llm-bc/lib/python3.9/site-packages/numpy/core/include -I/venv/llm-bc/include/python3.9 -c /venv/llm-bc/lib/python3.9/site-packages/mujoco_py/gl/osmesashim.c -o /venv/llm-bc/lib/python3.9/site-packages/mujoco_py/generated/_pyxbld_2.1.2.14_39_linuxcpuextensionbuilder/temp.linux-x86_64-cpython-39/venv/llm-bc/lib/python3.9/site-packages/mujoco_py/gl/osmesashim.o -fopenmp -w
+[2026-04-25 03:53:38,000][root][INFO] - gcc -pthread -B /venv/llm-bc/compiler_compat -shared -Wl,-rpath,/venv/llm-bc/lib -Wl,-rpath-link,/venv/llm-bc/lib -L/venv/llm-bc/lib -L/venv/llm-bc/lib -Wl,-rpath,/venv/llm-bc/lib -Wl,-rpath-link,/venv/llm-bc/lib -L/venv/llm-bc/lib /venv/llm-bc/lib/python3.9/site-packages/mujoco_py/generated/_pyxbld_2.1.2.14_39_linuxcpuextensionbuilder/temp.linux-x86_64-cpython-39/venv/llm-bc/lib/python3.9/site-packages/mujoco_py/cymj.o /venv/llm-bc/lib/python3.9/site-packages/mujoco_py/generated/_pyxbld_2.1.2.14_39_linuxcpuextensionbuilder/temp.linux-x86_64-cpython-39/venv/llm-bc/lib/python3.9/site-packages/mujoco_py/gl/osmesashim.o -L/root/.mujoco/mujoco210/bin -Wl,--enable-new-dtags,-rpath,/root/.mujoco/mujoco210/bin -lmujoco210 -lglewosmesa -lOSMesa -lGL -o /venv/llm-bc/lib/python3.9/site-packages/mujoco_py/generated/_pyxbld_2.1.2.14_39_linuxcpuextensionbuilder/lib.linux-x86_64-cpython-39/mujoco_py/cymj.cpython-39-x86_64-linux-gnu.so -fopenmp
+[2026-04-25 03:53:41,049][absl][INFO] - MUJOCO_GL=osmesa, attempting to import specified OpenGL backend.
+[2026-04-25 03:53:41,059][absl][INFO] - MuJoCo library version is: 2.3.7

2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/wandb/run-20260425_035342-iafad9fe/files/output.log

@@ -0,0 +1,3 @@
+Eval MetaworldLowdimRunner 1/6:   0%|                                                                                                                               | 0/30 [00:00<?, ?it/s]/workspace/LLM-BC/llmbc/common/llfbench_util.py:39: UserWarning: Creating a tensor from a list of numpy.ndarrays is extremely slow. Please consider converting the list to a single numpy.ndarray with numpy.array() before converting to a tensor. (Triggered internally at ../torch/csrc/utils/tensor_new.cpp:275.)
+  obs = torch.tensor(obs, dtype=torch.float32).unsqueeze(dim=0).to(device)
+Training epoch 809:   9%|██████▎                                                               | 76/846 [00:20<03:35,  3.57it/s, loss=0.0169]                                                           

2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/wandb/run-20260425_035342-iafad9fe/files/requirements.txt

@@ -0,0 +1,864 @@
+aiohappyeyeballs==2.6.1
+antlr4-python3-runtime==4.9.3
+asciitree==0.3.3
+async-timeout==5.0.1
+attrs==25.4.0
+certifi==2025.10.5
+charset-normalizer==3.4.4
+colorama==0.4.6
+dill==0.3.8
+docstring-parser==0.16
+eval_type_backport==0.2.2
+fasteners==0.20
+filelock==3.17.0
+frozenlist==1.8.0
+fsspec==2024.3.1
+gmpy2==2.2.1
+hf-xet==1.1.8
+idna==3.11
+llvmlite==0.39.1
+MarkupSafe==3.0.2
+mdurl==0.1.2
+mkl-service==2.4.0
+mpmath==1.3.0
+msgpack==1.1.1
+networkx==3.2.1
+packaging==25.0
+propcache==0.3.1
+psutil==7.0.0
+pycparser==2.23
+Pygments==2.19.1
+PyOpenGL==3.1.10
+PySocks==1.7.1
+tzdata==2025.3
+xxhash==3.5.0
+pytz==2025.2
+PyYAML==6.0.2
+regex==2025.9.1
+safetensors==0.5.3
+setuptools==80.9.0
+shtab==1.7.2
+six==1.17.0
+tqdm==4.67.1
+tensorboard==2.15.2
+uvloop==0.22.1
+wheel==0.45.1
+zipp==3.21.0
+tensorflow-estimator==2.15.0
+tensorboard-data-server==0.7.2
+google-auth-oauthlib==1.3.1
+tensorflow==2.15.1
+Jinja2==3.1.6
+markdown-it-py==2.2.0
+multiprocess==0.70.15
+numpy==1.23.5
+pillow==11.3.0
+pip==23.3.2
+python-dateutil==2.9.0.post0
+aiosignal==1.4.0
+brotlicffi==1.0.9.2
+hydra-core==1.2.0
+multidict==6.7.0
+pycares==4.10.0
+rich==14.2.0
+sympy==1.14.0
+typeguard==4.4.4
+aiodns==3.5.0
+urllib3==2.5.0
+yarl==1.18.0
+aiohttp==3.12.15
+pyarrow==21.0.0
+pyarrow-hotfix==0.7
+tokenizers==0.21.0
+av==10.0.0
+einops==0.4.1
+Bottleneck==1.4.2
+mkl_fft==1.3.11
+mkl_random==1.2.8
+numba==0.56.4
+numcodecs==0.12.1
+numexpr==2.10.1
+pandas==2.3.3
+zarr==2.12.0
+datasets==2.19.0
+transformers==4.47.1
+accelerate==1.0.1
+peft==0.14.0
+trl==0.11.4
+torch==2.2.2
+tabulate==0.9.0
+sentencepiece==0.2.0
+pyperclip==1.11.0
+PyOpenGL-accelerate==3.1.10
+pydub==0.25.1
+py-cpuinfo==9.0.0
+pure_eval==0.2.3
+ptyprocess==0.7.0
+nvidia-ml-py==13.595.45
+hjson==3.1.0
+gym-notices==0.1.0
+glfw==2.0.0
+Farama-Notifications==0.0.4
+websockets==15.0.1
+wcwidth==0.6.0
+uc-micro-py==1.0.3
+typing-inspection==0.4.2
+triton==2.2.0
+trimesh==4.11.5
+transforms3d==0.4.2
+traitlets==5.14.3
+threadpoolctl==3.6.0
+tenacity==9.1.2
+sniffio==1.3.1
+smmap==5.0.3
+setproctitle==1.3.7
+sentry-sdk==2.57.0
+semantic-version==2.10.0
+scipy==1.13.1
+rpds-py==0.27.1
+python-multipart==0.0.20
+pyparsing==3.3.2
+pynvml==13.0.1
+pydantic_core==2.41.5
+pycryptodomex==3.23.0
+pyasn1==0.6.3
+grpcio==1.80.0
+platformdirs==4.4.0
+pexpect==4.9.0
+parso==0.8.6
+orjson==3.11.5
+opencv-python==4.11.0.86
+stack-data==0.6.3
+omegaconf==2.2.1
+nvidia-nvtx-cu12==12.1.105
+nvidia-nvjitlink-cu12==12.9.86
+nvidia-nccl-cu12==2.19.3
+nvidia-curand-cu12==10.3.2.106
+nvidia-cufft-cu12==11.0.2.54
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cublas-cu12==12.1.3.1
+ninja==1.13.0
+narwhals==2.19.0
+lxml==6.0.3
+kiwisolver==1.4.7
+joblib==1.5.3
+jiter==0.13.0
+imageio-ffmpeg==0.6.0
+ImageIO==2.37.2
+h5py==3.14.0
+h11==0.16.0
+fonttools==4.60.2
+ffmpy==1.0.0
+fast_kinematics==0.2.2
+executing==2.2.1
+exceptiongroup==1.3.1
+docker-pycreds==0.4.0
+distro==1.9.0
+decorator==5.2.1
+dacite==1.9.2
+cycler==0.12.1
+contourpy==1.3.0
+cloudpickle==3.1.2
+click==8.1.8
+asttokens==3.0.1
+annotated-types==0.7.0
+annotated-doc==0.0.4
+aiofiles==25.1.0
+absl-py==2.3.1
+uvicorn==0.39.0
+tiktoken==0.8.0
+scikit-learn==1.6.1
+sapien==3.0.0b1
+referencing==0.36.2
+pydantic==2.12.5
+pyasn1_modules==0.4.2
+prompt_toolkit==3.0.52
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-cudnn-cu12==8.9.2.26
+nltk==3.9.2
+mujoco==2.3.7
+mdit-py-plugins==0.3.3
+matplotlib-inline==0.2.1
+matplotlib==3.7.5
+linkify-it-py==2.0.3
+jedi==0.19.2
+huggingface-hub==0.26.2
+httpcore==1.0.9
+gymnasium==0.29.1
+google-pasta==0.2.0
+gitdb==4.0.12
+cryptography==46.0.7
+blobfile==3.0.0
+anyio==4.12.1
+tyro==0.9.1
+toppra==0.6.6
+starlette==0.49.3
+seaborn==0.13.2
+rouge_score==0.1.2
+pynndescent==0.6.0
+nvidia-cusolver-cu12==11.4.5.107
+jsonschema-specifications==2025.9.1
+ipython==8.18.1
+httpx==0.28.1
+gymnasium-robotics==1.2.0
+google-auth==2.49.2
+GitPython==3.1.46
+diffusers==0.31.0
+wandb==0.18.6
+umap-learn==0.5.9.post2
+requests-oauthlib==2.0.0
+openai==2.8.1
+mplib==0.1.1
+jsonschema==4.25.1
+gradio_client==0.2.9
+google-genai==1.47.0
+fastapi==0.128.8
+torchvision==0.17.2
+torchaudio==2.2.2
+sentence-transformers==3.2.1
+pytorch-seed==0.2.0
+deepspeed==0.16.1
+bitsandbytes==0.45.0
+altair==6.0.0
+timm==1.0.22
+gradio==3.36.1
+evaluate==0.4.3
+arm_pytorch_utilities==0.5.0
+tensorflow-io-gcs-filesystem==0.37.1
+Markdown==3.9
+pybullet==3.2.6
+wrapt==1.14.2
+tensorflow-probability==0.23.0
+parse==1.19.1
+dm-tree==0.1.8
+termcolor==3.1.0
+requests==2.32.0
+cffi==2.0.0
+opt_einsum==3.4.0
+labmaze==1.0.6
+typing_extensions==4.15.0
+importlib-resources==5.13.0
+importlib-metadata==5.2.0
+FLAML==2.3.6
+dm-env==1.6
+diskcache==5.6.3
+Cython==0.29.37
+mujoco-py==2.1.2.14
+jaxlib==0.4.30
+dm-control==1.0.14
+cmudict==1.0.13
+syllables==1.0.9
+metaworld==2.0.0
+jax==0.4.30
+highway-env==1.9.1
+gym_bandits==0.0.2
+d4rl==1.1
+pyautogen==0.1.0
+protobuf==4.25.9
+tf-agents==0.19.0
+oauthlib==3.3.1
+pytorch-kinematics==0.7.5
+pygame==2.6.1
+gin-config==0.5.0
+mani_skill==3.0.0b20
+tifffile==2024.8.30
+Shapely==1.8.4
+PyWavelets==1.6.0
+Werkzeug==3.1.8
+astunparse==1.6.3
+scikit-image==0.19.3
+pymunk==6.2.1
+ml-dtypes==0.3.2
+gast==0.7.0
+stable-baselines3==2.2.1
+keras==2.15.0
+libclang==18.1.1
+flatbuffers==25.12.19
+gym==0.23.1
+llmbc==0.0.0
+mjrl==1.0.0
+llfbench==0.1.0
+aiohappyeyeballs==2.6.1
+antlr4-python3-runtime==4.9.3
+asciitree==0.3.3
+async-timeout==5.0.1
+attrs==25.4.0
+certifi==2025.10.5
+charset-normalizer==3.4.4
+colorama==0.4.6
+dill==0.3.8
+docstring-parser==0.16
+eval_type_backport==0.2.2
+fasteners==0.20
+filelock==3.17.0
+frozenlist==1.8.0
+fsspec==2024.3.1
+gmpy2==2.2.1
+hf-xet==1.1.8
+idna==3.11
+llvmlite==0.39.1
+MarkupSafe==3.0.2
+mdurl==0.1.2
+mkl-service==2.4.0
+mpmath==1.3.0
+msgpack==1.1.1
+networkx==3.2.1
+packaging==25.0
+propcache==0.3.1
+psutil==7.0.0
+pycparser==2.23
+Pygments==2.19.1
+PyOpenGL==3.1.10
+PySocks==1.7.1
+tzdata==2025.3
+xxhash==3.5.0
+pytz==2025.2
+PyYAML==6.0.2
+regex==2025.9.1
+safetensors==0.5.3
+setuptools==80.9.0
+shtab==1.7.2
+six==1.17.0
+tqdm==4.67.1
+tensorboard==2.15.2
+uvloop==0.22.1
+wheel==0.45.1
+zipp==3.21.0
+tensorflow-estimator==2.15.0
+tensorboard-data-server==0.7.2
+google-auth-oauthlib==1.3.1
+tensorflow==2.15.1
+Jinja2==3.1.6
+markdown-it-py==2.2.0
+multiprocess==0.70.15
+numpy==1.23.5
+pillow==11.3.0
+pip==23.3.2
+python-dateutil==2.9.0.post0
+aiosignal==1.4.0
+brotlicffi==1.0.9.2
+hydra-core==1.2.0
+multidict==6.7.0
+pycares==4.10.0
+rich==14.2.0
+sympy==1.14.0
+typeguard==4.4.4
+aiodns==3.5.0
+urllib3==2.5.0
+yarl==1.18.0
+aiohttp==3.12.15
+pyarrow==21.0.0
+pyarrow-hotfix==0.7
+tokenizers==0.21.0
+av==10.0.0
+einops==0.4.1
+Bottleneck==1.4.2
+mkl_fft==1.3.11
+mkl_random==1.2.8
+numba==0.56.4
+numcodecs==0.12.1
+numexpr==2.10.1
+pandas==2.3.3
+zarr==2.12.0
+datasets==2.19.0
+transformers==4.47.1
+accelerate==1.0.1
+peft==0.14.0
+trl==0.11.4
+torch==2.2.2
+tabulate==0.9.0
+sentencepiece==0.2.0
+pyperclip==1.11.0
+PyOpenGL-accelerate==3.1.10
+pydub==0.25.1
+py-cpuinfo==9.0.0
+pure_eval==0.2.3
+ptyprocess==0.7.0
+nvidia-ml-py==13.595.45
+hjson==3.1.0
+gym-notices==0.1.0
+glfw==2.0.0
+Farama-Notifications==0.0.4
+websockets==15.0.1
+wcwidth==0.6.0
+uc-micro-py==1.0.3
+typing-inspection==0.4.2
+triton==2.2.0
+trimesh==4.11.5
+transforms3d==0.4.2
+traitlets==5.14.3
+threadpoolctl==3.6.0
+tenacity==9.1.2
+sniffio==1.3.1
+smmap==5.0.3
+setproctitle==1.3.7
+sentry-sdk==2.57.0
+semantic-version==2.10.0
+scipy==1.13.1
+rpds-py==0.27.1
+python-multipart==0.0.20
+pyparsing==3.3.2
+pynvml==13.0.1
+pydantic_core==2.41.5
+pycryptodomex==3.23.0
+pyasn1==0.6.3
+grpcio==1.80.0
+platformdirs==4.4.0
+pexpect==4.9.0
+parso==0.8.6
+orjson==3.11.5
+opencv-python==4.11.0.86
+stack-data==0.6.3
+omegaconf==2.2.1
+nvidia-nvtx-cu12==12.1.105
+nvidia-nvjitlink-cu12==12.9.86
+nvidia-nccl-cu12==2.19.3
+nvidia-curand-cu12==10.3.2.106
+nvidia-cufft-cu12==11.0.2.54
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cublas-cu12==12.1.3.1
+ninja==1.13.0
+narwhals==2.19.0
+lxml==6.0.3
+kiwisolver==1.4.7
+joblib==1.5.3
+jiter==0.13.0
+imageio-ffmpeg==0.6.0
+ImageIO==2.37.2
+h5py==3.14.0
+h11==0.16.0
+fonttools==4.60.2
+ffmpy==1.0.0
+fast_kinematics==0.2.2
+executing==2.2.1
+exceptiongroup==1.3.1
+docker-pycreds==0.4.0
+distro==1.9.0
+decorator==5.2.1
+dacite==1.9.2
+cycler==0.12.1
+contourpy==1.3.0
+cloudpickle==3.1.2
+click==8.1.8
+asttokens==3.0.1
+annotated-types==0.7.0
+annotated-doc==0.0.4
+aiofiles==25.1.0
+absl-py==2.3.1
+uvicorn==0.39.0
+tiktoken==0.8.0
+scikit-learn==1.6.1
+sapien==3.0.0b1
+referencing==0.36.2
+pydantic==2.12.5
+pyasn1_modules==0.4.2
+prompt_toolkit==3.0.52
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-cudnn-cu12==8.9.2.26
+nltk==3.9.2
+mujoco==2.3.7
+mdit-py-plugins==0.3.3
+matplotlib-inline==0.2.1
+matplotlib==3.7.5
+linkify-it-py==2.0.3
+jedi==0.19.2
+huggingface-hub==0.26.2
+httpcore==1.0.9
+gymnasium==0.29.1
+google-pasta==0.2.0
+gitdb==4.0.12
+cryptography==46.0.7
+blobfile==3.0.0
+anyio==4.12.1
+tyro==0.9.1
+toppra==0.6.6
+starlette==0.49.3
+seaborn==0.13.2
+rouge_score==0.1.2
+pynndescent==0.6.0
+nvidia-cusolver-cu12==11.4.5.107
+jsonschema-specifications==2025.9.1
+ipython==8.18.1
+httpx==0.28.1
+gymnasium-robotics==1.2.0
+google-auth==2.49.2
+GitPython==3.1.46
+diffusers==0.31.0
+wandb==0.18.6
+umap-learn==0.5.9.post2
+requests-oauthlib==2.0.0
+openai==2.8.1
+mplib==0.1.1
+jsonschema==4.25.1
+gradio_client==0.2.9
+google-genai==1.47.0
+fastapi==0.128.8
+torchvision==0.17.2
+torchaudio==2.2.2
+sentence-transformers==3.2.1
+pytorch-seed==0.2.0
+deepspeed==0.16.1
+bitsandbytes==0.45.0
+altair==6.0.0
+timm==1.0.22
+gradio==3.36.1
+evaluate==0.4.3
+arm_pytorch_utilities==0.5.0
+tensorflow-io-gcs-filesystem==0.37.1
+Markdown==3.9
+pybullet==3.2.6
+wrapt==1.14.2
+tensorflow-probability==0.23.0
+parse==1.19.1
+dm-tree==0.1.8
+termcolor==3.1.0
+requests==2.32.0
+cffi==2.0.0
+opt_einsum==3.4.0
+labmaze==1.0.6
+typing_extensions==4.15.0
+importlib-resources==5.13.0
+importlib-metadata==5.2.0
+FLAML==2.3.6
+dm-env==1.6
+diskcache==5.6.3
+Cython==0.29.37
+mujoco-py==2.1.2.14
+jaxlib==0.4.30
+dm-control==1.0.14
+cmudict==1.0.13
+syllables==1.0.9
+metaworld==2.0.0
+jax==0.4.30
+highway-env==1.9.1
+gym_bandits==0.0.2
+d4rl==1.1
+pyautogen==0.1.0
+protobuf==4.25.9
+tf-agents==0.19.0
+oauthlib==3.3.1
+pytorch-kinematics==0.7.5
+pygame==2.6.1
+gin-config==0.5.0
+mani_skill==3.0.0b20
+tifffile==2024.8.30
+Shapely==1.8.4
+PyWavelets==1.6.0
+Werkzeug==3.1.8
+astunparse==1.6.3
+scikit-image==0.19.3
+pymunk==6.2.1
+ml-dtypes==0.3.2
+gast==0.7.0
+stable-baselines3==2.2.1
+keras==2.15.0
+libclang==18.1.1
+flatbuffers==25.12.19
+gym==0.23.1
+llmbc==0.0.0
+mjrl==1.0.0
+llfbench==0.1.0
+llmbc==0.0.0
+llmbc==0.0.0
+aiohappyeyeballs==2.6.1
+antlr4-python3-runtime==4.9.3
+asciitree==0.3.3
+async-timeout==5.0.1
+attrs==25.4.0
+certifi==2025.10.5
+charset-normalizer==3.4.4
+colorama==0.4.6
+dill==0.3.8
+docstring-parser==0.16
+eval_type_backport==0.2.2
+fasteners==0.20
+filelock==3.17.0
+frozenlist==1.8.0
+fsspec==2024.3.1
+gmpy2==2.2.1
+hf-xet==1.1.8
+idna==3.11
+llvmlite==0.39.1
+MarkupSafe==3.0.2
+mdurl==0.1.2
+mkl-service==2.4.0
+mpmath==1.3.0
+msgpack==1.1.1
+networkx==3.2.1
+packaging==25.0
+propcache==0.3.1
+psutil==7.0.0
+pycparser==2.23
+Pygments==2.19.1
+PyOpenGL==3.1.10
+PySocks==1.7.1
+tzdata==2025.3
+xxhash==3.5.0
+pytz==2025.2
+PyYAML==6.0.2
+regex==2025.9.1
+safetensors==0.5.3
+setuptools==80.9.0
+shtab==1.7.2
+six==1.17.0
+tqdm==4.67.1
+tensorboard==2.15.2
+uvloop==0.22.1
+wheel==0.45.1
+zipp==3.21.0
+tensorflow-estimator==2.15.0
+tensorboard-data-server==0.7.2
+google-auth-oauthlib==1.3.1
+tensorflow==2.15.1
+Jinja2==3.1.6
+markdown-it-py==2.2.0
+multiprocess==0.70.15
+numpy==1.23.5
+pillow==11.3.0
+pip==23.3.2
+python-dateutil==2.9.0.post0
+aiosignal==1.4.0
+brotlicffi==1.0.9.2
+hydra-core==1.2.0
+multidict==6.7.0
+pycares==4.10.0
+rich==14.2.0
+sympy==1.14.0
+typeguard==4.4.4
+aiodns==3.5.0
+urllib3==2.5.0
+yarl==1.18.0
+aiohttp==3.12.15
+pyarrow==21.0.0
+pyarrow-hotfix==0.7
+tokenizers==0.21.0
+av==10.0.0
+einops==0.4.1
+Bottleneck==1.4.2
+mkl_fft==1.3.11
+mkl_random==1.2.8
+numba==0.56.4
+numcodecs==0.12.1
+numexpr==2.10.1
+pandas==2.3.3
+zarr==2.12.0
+datasets==2.19.0
+transformers==4.47.1
+accelerate==1.0.1
+peft==0.14.0
+trl==0.11.4
+torch==2.2.2
+tabulate==0.9.0
+sentencepiece==0.2.0
+pyperclip==1.11.0
+PyOpenGL-accelerate==3.1.10
+pydub==0.25.1
+py-cpuinfo==9.0.0
+pure_eval==0.2.3
+ptyprocess==0.7.0
+nvidia-ml-py==13.595.45
+hjson==3.1.0
+gym-notices==0.1.0
+glfw==2.0.0
+Farama-Notifications==0.0.4
+websockets==15.0.1
+wcwidth==0.6.0
+uc-micro-py==1.0.3
+typing-inspection==0.4.2
+triton==2.2.0
+trimesh==4.11.5
+transforms3d==0.4.2
+traitlets==5.14.3
+threadpoolctl==3.6.0
+tenacity==9.1.2
+sniffio==1.3.1
+smmap==5.0.3
+setproctitle==1.3.7
+sentry-sdk==2.57.0
+semantic-version==2.10.0
+scipy==1.13.1
+rpds-py==0.27.1
+python-multipart==0.0.20
+pyparsing==3.3.2
+pynvml==13.0.1
+pydantic_core==2.41.5
+pycryptodomex==3.23.0
+pyasn1==0.6.3
+grpcio==1.80.0
+platformdirs==4.4.0
+pexpect==4.9.0
+parso==0.8.6
+orjson==3.11.5
+opencv-python==4.11.0.86
+stack-data==0.6.3
+omegaconf==2.2.1
+nvidia-nvtx-cu12==12.1.105
+nvidia-nvjitlink-cu12==12.9.86
+nvidia-nccl-cu12==2.19.3
+nvidia-curand-cu12==10.3.2.106
+nvidia-cufft-cu12==11.0.2.54
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cublas-cu12==12.1.3.1
+ninja==1.13.0
+narwhals==2.19.0
+lxml==6.0.3
+kiwisolver==1.4.7
+joblib==1.5.3
+jiter==0.13.0
+imageio-ffmpeg==0.6.0
+ImageIO==2.37.2
+h5py==3.14.0
+h11==0.16.0
+fonttools==4.60.2
+ffmpy==1.0.0
+fast_kinematics==0.2.2
+executing==2.2.1
+exceptiongroup==1.3.1
+docker-pycreds==0.4.0
+distro==1.9.0
+decorator==5.2.1
+dacite==1.9.2
+cycler==0.12.1
+contourpy==1.3.0
+cloudpickle==3.1.2
+click==8.1.8
+asttokens==3.0.1
+annotated-types==0.7.0
+annotated-doc==0.0.4
+aiofiles==25.1.0
+absl-py==2.3.1
+uvicorn==0.39.0
+tiktoken==0.8.0
+scikit-learn==1.6.1
+sapien==3.0.0b1
+referencing==0.36.2
+pydantic==2.12.5
+pyasn1_modules==0.4.2
+prompt_toolkit==3.0.52
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-cudnn-cu12==8.9.2.26
+nltk==3.9.2
+mujoco==2.3.7
+mdit-py-plugins==0.3.3
+matplotlib-inline==0.2.1
+matplotlib==3.7.5
+linkify-it-py==2.0.3
+jedi==0.19.2
+huggingface-hub==0.26.2
+httpcore==1.0.9
+gymnasium==0.29.1
+google-pasta==0.2.0
+gitdb==4.0.12
+cryptography==46.0.7
+blobfile==3.0.0
+anyio==4.12.1
+tyro==0.9.1
+toppra==0.6.6
+starlette==0.49.3
+seaborn==0.13.2
+rouge_score==0.1.2
+pynndescent==0.6.0
+nvidia-cusolver-cu12==11.4.5.107
+jsonschema-specifications==2025.9.1
+ipython==8.18.1
+httpx==0.28.1
+gymnasium-robotics==1.2.0
+google-auth==2.49.2
+GitPython==3.1.46
+diffusers==0.31.0
+wandb==0.18.6
+umap-learn==0.5.9.post2
+requests-oauthlib==2.0.0
+openai==2.8.1
+mplib==0.1.1
+jsonschema==4.25.1
+gradio_client==0.2.9
+google-genai==1.47.0
+fastapi==0.128.8
+torchvision==0.17.2
+torchaudio==2.2.2
+sentence-transformers==3.2.1
+pytorch-seed==0.2.0
+deepspeed==0.16.1
+bitsandbytes==0.45.0
+altair==6.0.0
+timm==1.0.22
+gradio==3.36.1
+evaluate==0.4.3
+arm_pytorch_utilities==0.5.0
+tensorflow-io-gcs-filesystem==0.37.1
+Markdown==3.9
+pybullet==3.2.6
+wrapt==1.14.2
+tensorflow-probability==0.23.0
+parse==1.19.1
+dm-tree==0.1.8
+termcolor==3.1.0
+requests==2.32.0
+cffi==2.0.0
+opt_einsum==3.4.0
+labmaze==1.0.6
+typing_extensions==4.15.0
+importlib-resources==5.13.0
+importlib-metadata==5.2.0
+FLAML==2.3.6
+dm-env==1.6
+diskcache==5.6.3
+Cython==0.29.37
+mujoco-py==2.1.2.14
+jaxlib==0.4.30
+dm-control==1.0.14
+cmudict==1.0.13
+syllables==1.0.9
+metaworld==2.0.0
+jax==0.4.30
+highway-env==1.9.1
+gym_bandits==0.0.2
+d4rl==1.1
+pyautogen==0.1.0
+protobuf==4.25.9
+tf-agents==0.19.0
+oauthlib==3.3.1
+pytorch-kinematics==0.7.5
+pygame==2.6.1
+gin-config==0.5.0
+mani_skill==3.0.0b20
+tifffile==2024.8.30
+Shapely==1.8.4
+PyWavelets==1.6.0
+Werkzeug==3.1.8
+astunparse==1.6.3
+scikit-image==0.19.3
+pymunk==6.2.1
+ml-dtypes==0.3.2
+gast==0.7.0
+stable-baselines3==2.2.1
+keras==2.15.0
+libclang==18.1.1
+flatbuffers==25.12.19
+gym==0.23.1
+llmbc==0.0.0
+mjrl==1.0.0
+llfbench==0.1.0
+autocommand==2.2.2
+backports.tarfile==1.2.0
+importlib_metadata==8.0.0
+inflect==7.3.1
+jaraco.collections==5.1.0
+jaraco.context==5.3.0
+jaraco.functools==4.0.1
+jaraco.text==3.12.1
+more-itertools==10.3.0
+packaging==24.2
+platformdirs==4.2.2
+tomli==2.0.1
+typeguard==4.3.0
+typing_extensions==4.12.2
+wheel==0.45.1
+zipp==3.19.2

2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/wandb/run-20260425_035342-iafad9fe/files/wandb-metadata.json

@@ -0,0 +1,56 @@
+{
+  "os": "Linux-5.15.0-168-generic-x86_64-with-glibc2.39",
+  "python": "3.9.25",
+  "startedAt": "2026-04-25T03:53:42.012724Z",
+  "args": [
+    "--config-path",
+    "config/mix_data",
+    "--config-name",
+    "llmdp_sweep-v2_mix(csp3).yaml",
+    "training.seed=43"
+  ],
+  "program": "/workspace/LLM-BC/./train.py",
+  "codePath": "train.py",
+  "git": {
+    "remote": "https://github.com/CHYang25/LLM-BC.git",
+    "commit": "0131b06a1134730d8afeb66e613e0eb72963e3c4"
+  },
+  "email": "chris920325@gmail.com",
+  "root": "/workspace/LLM-BC/data/outputs/2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2",
+  "host": "c8ef52649900",
+  "username": "root",
+  "executable": "/venv/llm-bc/bin/python3",
+  "codePathLocal": "train.py",
+  "cpu_count": 72,
+  "cpu_count_logical": 144,
+  "gpu": "NVIDIA GeForce RTX 4090",
+  "gpu_count": 2,
+  "disk": {
+    "/": {
+      "total": "545460846592",
+      "used": "476012060672"
+    }
+  },
+  "memory": {
+    "total": "810942898176"
+  },
+  "cpu": {
+    "count": 72,
+    "countLogical": 144
+  },
+  "gpu_nvidia": [
+    {
+      "name": "NVIDIA GeForce RTX 4090",
+      "memoryTotal": "25757220864",
+      "cudaCores": 16384,
+      "architecture": "Ada"
+    },
+    {
+      "name": "NVIDIA GeForce RTX 4090",
+      "memoryTotal": "25757220864",
+      "cudaCores": 16384,
+      "architecture": "Ada"
+    }
+  ],
+  "cudaVersion": "12.8"
+}

2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/wandb/run-20260425_035342-iafad9fe/run-iafad9fe.wandb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6da001030f3221d3a3c1b60f200c44d2cbb4b14decb3b83d631d449dbf01c95d
+size 682819584

2026.04.25/03.52.11_train_llm_diffusion_unet_lowdim_sweep-v2/wandb/wandb-resume.json

@@ -0,0 +1 @@
+{"run_id": "iafad9fe"}

2026.04.25/22.26.21_train_llm_diffusion_unet_lowdim_box-close-v2/.hydra/config.yaml

@@ -0,0 +1,193 @@
+name: train_llm_diffusion_unet_lowdim
+_target_: llmbc.workspace.train_llm_diffusion_unet_lowdim_workspace.TrainLLMDiffusionUnetLowdimWorkspace
+obs_dim: ${task.obs_dim}
+action_dim: ${task.action_dim}
+task_name: ${task.name}
+exp_name: default
+model_name: ${llm.name}
+horizon: 1
+n_obs_steps: 1
+n_action_steps: 1
+n_latency_steps: 0
+past_action_visible: false
+keypoint_visible_rate: 1.0
+obs_as_local_cond: false
+obs_as_global_cond: true
+pred_action_steps_only: false
+llm_orig_expert_feedback: true
+llm_do_sample: false
+policy:
+  _target_: llmbc.policy.llm_diffusion_unet_lowdim_policy.LLMDiffusionUnetLowdimPolicy
+  model:
+    _target_: llmbc.model.diffusion.conditional_unet1d.ConditionalUnet1D
+    input_dim: '${eval: ${task.action_dim} if ${obs_as_local_cond} or ${obs_as_global_cond}
+      else ${task.obs_dim} + ${task.action_dim}}'
+    local_cond_dim: '${eval: ${task.obs_dim} if ${obs_as_local_cond} else None}'
+    global_cond_dim: '${eval: ${task.obs_dim}*${n_obs_steps} if ${obs_as_global_cond}
+      else None}'
+    diffusion_step_embed_dim: 256
+    down_dims:
+    - 256
+    - 512
+    - 1024
+    kernel_size: 5
+    n_groups: 8
+    cond_predict_scale: true
+  noise_scheduler:
+    _target_: diffusers.schedulers.scheduling_ddpm.DDPMScheduler
+    num_train_timesteps: 100
+    beta_start: 0.0001
+    beta_end: 0.02
+    beta_schedule: squaredcos_cap_v2
+    variance_type: fixed_small
+    clip_sample: true
+    prediction_type: epsilon
+  horizon: ${horizon}
+  obs_dim: ${obs_dim}
+  action_dim: ${action_dim}
+  n_action_steps: ${eval:'${n_action_steps}+${n_latency_steps}'}
+  n_obs_steps: ${n_obs_steps}
+  num_inference_steps: 100
+  obs_as_local_cond: ${obs_as_local_cond}
+  obs_as_global_cond: ${obs_as_global_cond}
+  pred_action_steps_only: ${pred_action_steps_only}
+  oa_step_convention: true
+  llm_discriminator:
+    _target_: llmbc.discriminator.llm_ce_discriminator.LLMCEDiscriminator
+    task_id: ${task_name}
+    llm_translator:
+      _target_: llmbc.translator.llm_translator.LLMTranslator
+      cfg: ${llm}
+      obs_dim: ${obs_dim}
+      action_dim: ${action_dim}
+      horizon: ${horizon}
+      n_obs_steps: ${n_obs_steps}
+      n_action_steps: ${eval:'${n_action_steps}+${n_latency_steps}'}
+  loss_dp_weight: 1.0
+  loss_llm_weight: 0.01
+  normalize_llm_loss: true
+  reweight_llm_loss: true
+ema:
+  _target_: llmbc.model.diffusion.ema_model.EMAModel
+  update_after_step: 0
+  inv_gamma: 1.0
+  power: 0.75
+  min_value: 0.0
+  max_value: 0.9999
+dataloader:
+  batch_size: 16
+  num_workers: 0
+  shuffle: true
+  pin_memory: false
+  persistent_workers: false
+val_dataloader:
+  batch_size: 16
+  num_workers: 0
+  shuffle: false
+  pin_memory: false
+  persistent_workers: false
+optimizer:
+  _target_: torch.optim.AdamW
+  lr: 0.0001
+  betas:
+  - 0.95
+  - 0.999
+  eps: 1.0e-08
+  weight_decay: 1.0e-06
+training:
+  device: cuda:0
+  seed: 44
+  debug: false
+  resume: true
+  lr_scheduler: cosine
+  lr_warmup_steps: 500
+  num_epochs: 1001
+  gradient_accumulate_every: 16
+  use_ema: true
+  rollout_every: 5
+  checkpoint_every: 5
+  val_every: 1
+  sample_every: 5
+  max_train_steps: null
+  max_val_steps: null
+  tqdm_interval_sec: 1.0
+logging:
+  project: ${task.name}-training-mix
+  resume: true
+  mode: online
+  name: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
+  tags:
+  - ${name}
+  - ${task_name}
+  - ${exp_name}
+  id: null
+  group: null
+checkpoint:
+  topk:
+    monitor_key: test_success_rate
+    mode: max
+    k: 5
+    format_str: epoch={epoch:04d}-test_success_rate={test_success_rate:.3f}.ckpt
+  save_last_ckpt: true
+  save_last_snapshot: false
+multi_run:
+  run_dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+  wandb_name_base: ${now:%Y.%m.%d-%H.%M.%S}_${name}_${task_name}
+task:
+  name: box-close-v2
+  obs_dim: 9
+  action_dim: 4
+  env_runner:
+    _target_: llmbc.env_runner.metaworld_lowdim_runner.MetaworldLowdimRunner
+    env_name: llf-metaworld-box-close-v2
+    n_train: 10
+    n_test: 50
+    n_envs: 10
+    max_steps: 30
+    n_obs_steps: ${n_obs_steps}
+    n_action_steps: ${n_action_steps}
+    instruction_type: b
+    feedback_type:
+    - hp
+    - hn
+    - fp
+    visual: false
+    discount: 0.9
+  dataset:
+    _target_: llmbc.dataset.metaworld_lowdim_dataset.MetaworldLowdimDataset
+    data_path: datasets/box-close-v2-mix(csp3).pt
+    data_path2: datasets/box-close-v2.pt
+    horizon: ${horizon}
+    pad_before: ${eval:'${n_obs_steps}-1'}
+    pad_after: ${eval:'${n_action_steps}-1'}
+    obs_eef_target: true
+    use_manual_normalizer: false
+    val_ratio: 0.1
+    dummy_normalizer: false
+  instructor:
+    _target_: llmbc.translator.instructor.metaworld_instructor.box_close_v2_instructor.BoxCloseV2Instructor
+llm:
+  name: HuggingFaceTB/SmolLM2-135M-Instruct
+  model_name: SmolLM2-135M-Instruct
+  config_target: llmbc.model.llm.llama_lowdim_model.LowdimLlamaConfig
+  causal_lm_target: llmbc.model.llm.llama_lowdim_model.LowdimLlamaForCausalLM
+  use_quantization: false
+  use_joint_mlp_projector: true
+  llm_mode: ete-finetuned
+  finetune_mode: orig
+  checkpoint: data/outputs/2025.10.22/23.44.28_train_llm_lowdim_box-close-v2/HuggingFaceTB/SmolLM2-135M-Instruct-finetuned-box-close-v2/checkpoint-5890
+  max_length: 100
+  lora_config:
+    r: 32
+    lora_alpha: 64
+    lora_dropout: 0.05
+    bias: none
+    task_type: CAUSAL_LM
+  prompter:
+    _target_: llmbc.translator.prompter.smollm2_prompter.SmolLM2Prompter
+    use_joint_mlp_projector: true
+  hydra:
+    job:
+      override_dirname: ${model_name}
+    run:
+      dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${model_name}

2026.04.25/22.26.21_train_llm_diffusion_unet_lowdim_box-close-v2/.hydra/hydra.yaml

@@ -0,0 +1,155 @@
+hydra:
+  run:
+    dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+  sweep:
+    dir: data/outputs/${now:%Y.%m.%d}/${now:%H.%M.%S}_${name}_${task_name}
+    subdir: ${hydra.job.num}
+  launcher:
+    _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
+  sweeper:
+    _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
+    max_batch_size: null
+    params: null
+  help:
+    app_name: ${hydra.job.name}
+    header: '${hydra.help.app_name} is powered by Hydra.
+
+      '
+    footer: 'Powered by Hydra (https://hydra.cc)
+
+      Use --hydra-help to view Hydra specific help
+
+      '
+    template: '${hydra.help.header}
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (group=option)
+
+
+      $APP_CONFIG_GROUPS
+
+
+      == Config ==
+
+      Override anything in the config (foo.bar=value)
+
+
+      $CONFIG
+
+
+      ${hydra.help.footer}
+
+      '
+  hydra_help:
+    template: 'Hydra (${hydra.runtime.version})
+
+      See https://hydra.cc for more info.
+
+
+      == Flags ==
+
+      $FLAGS_HELP
+
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (For example, append hydra/job_logging=disabled
+      to command line)
+
+
+      $HYDRA_CONFIG_GROUPS
+
+
+      Use ''--cfg hydra'' to Show the Hydra config.
+
+      '
+    hydra_help: ???
+  hydra_logging:
+    version: 1
+    formatters:
+      simple:
+        format: '[%(asctime)s][HYDRA] %(message)s'
+    handlers:
+      console:
+        class: logging.StreamHandler
+        formatter: simple
+        stream: ext://sys.stdout
+    root:
+      level: INFO
+      handlers:
+      - console
+    loggers:
+      logging_example:
+        level: DEBUG
+    disable_existing_loggers: false
+  job_logging:
+    version: 1
+    formatters:
+      simple:
+        format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
+    handlers:
+      console:
+        class: logging.StreamHandler
+        formatter: simple
+        stream: ext://sys.stdout
+      file:
+        class: logging.FileHandler
+        formatter: simple
+        filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
+    root:
+      level: INFO
+      handlers:
+      - console
+      - file
+    disable_existing_loggers: false
+  env: {}
+  mode: RUN
+  searchpath: []
+  callbacks: {}
+  output_subdir: .hydra
+  overrides:
+    hydra:
+    - hydra.mode=RUN
+    task:
+    - training.seed=44
+  job:
+    name: train
+    chdir: null
+    override_dirname: training.seed=44
+    id: ???
+    num: ???
+    config_name: llmdp_box-close-v2_mix(csp3).yaml
+    env_set: {}
+    env_copy: []
+    config:
+      override_dirname:
+        kv_sep: '='
+        item_sep: ','
+        exclude_keys: []
+  runtime:
+    version: 1.2.0
+    version_base: '1.2'
+    cwd: /workspace/LLM-BC
+    config_sources:
+    - path: hydra.conf
+      schema: pkg
+      provider: hydra
+    - path: /workspace/LLM-BC/config/mix_data
+      schema: file
+      provider: main
+    - path: ''
+      schema: structured
+      provider: schema
+    output_dir: /workspace/LLM-BC/data/outputs/2026.04.25/22.26.21_train_llm_diffusion_unet_lowdim_box-close-v2
+    choices:
+      hydra/env: default
+      hydra/callbacks: null
+      hydra/job_logging: default
+      hydra/hydra_logging: default
+      hydra/hydra_help: default
+      hydra/help: default
+      hydra/sweeper: basic
+      hydra/launcher: basic
+      hydra/output: default
+  verbose: false

2026.04.25/22.26.21_train_llm_diffusion_unet_lowdim_box-close-v2/.hydra/overrides.yaml

@@ -0,0 +1 @@
+- training.seed=44

2026.04.25/22.26.21_train_llm_diffusion_unet_lowdim_box-close-v2/checkpoints/epoch=0670-test_success_rate=0.900.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3bde11b59dbed7ab40a309f726641cd217af1a9c1b97281a3fac9a57343f5502
+size 1040498994

2026.04.25/22.26.21_train_llm_diffusion_unet_lowdim_box-close-v2/checkpoints/epoch=0710-test_success_rate=0.880.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:601437a6e561e9a77db79a745277f071e639aa0244f954ea21bce6eb2b44f78f
+size 1040498994

2026.04.25/22.26.21_train_llm_diffusion_unet_lowdim_box-close-v2/checkpoints/epoch=0785-test_success_rate=0.920.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8a091508b3d894bdc5304078dfb7d0468ed0a904741fe13578da3233c4b2822a
+size 1040498994

2026.04.25/22.26.21_train_llm_diffusion_unet_lowdim_box-close-v2/checkpoints/epoch=0885-test_success_rate=0.920.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:848fffec505da0c64134d0088026241198e760ae6e8c08ca4a606d4b2df055a7
+size 1040498994

2026.04.25/22.26.21_train_llm_diffusion_unet_lowdim_box-close-v2/checkpoints/epoch=0945-test_success_rate=0.900.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fbf301c365bc19befb9fb0898d8c1182dcc1b6e0af3ce3a38b0a415783a5ba4a
+size 1040498994

2026.04.25/22.26.21_train_llm_diffusion_unet_lowdim_box-close-v2/checkpoints/latest.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2324c5305575f9a2ad7464f437d0c44a2121b5c254614f0a090240e6ed2f8626
+size 1040498994