Views
No views yet
1pip install huggingface_hub
2huggingface-cli download denkiwakame/Qwen3.5-4B-FFT-LAP-UR5e-PyAV --local-dir ./Qwen3.5-4B-FFT-LAP-UR5e-PyAV1import pickle
2import torch
3from transformers import Qwen2_5_VLForConditionalGeneration, Qwen2_5_VLProcessor
4
5ckpt_dir = "./Qwen3.5-4B-FFT-LAP-UR5e-PyAV"
6
7model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
8 f"{ckpt_dir}/model_final", torch_dtype=torch.bfloat16, device_map="auto",
9)
10processor = Qwen2_5_VLProcessor.from_pretrained(f"{ckpt_dir}/model_final")
11
12# Load dataset stats (required for action denormalization)
13with open(f"{ckpt_dir}/dataset_stats.pkl", "rb") as f:
14 dataset_stats = pickle.load(f)1from rv_train.train import get_pretrained_model
2
3model, cfg = get_pretrained_model("./Qwen3.5-4B-FFT-LAP-UR5e-PyAV", device=0)
4model.eval()dataset_stats.pkl1import pickle
2
3with open("dataset_stats.pkl", "rb") as f:
4 stats = pickle.load(f)
5# stats contains mean/std for action dimensionsmain holds the recommended/final weights.
Earlier training-step snapshots are published as branches named step-<global_step> (e.g., step-17000, step-18000).
Load any of them by passing revision=:1# Download a specific revision
2huggingface-cli download denkiwakame/Qwen3.5-4B-FFT-LAP-UR5e-PyAV --revision step-18000 --local-dir ./Qwen3.5-4B-FFT-LAP-UR5e-PyAV-step-18000
3
4# Or load directly via transformers
5Qwen2_5_VLForConditionalGeneration.from_pretrained(
6 "denkiwakame/Qwen3.5-4B-FFT-LAP-UR5e-PyAV",
7 revision="step-18000",
8 subfolder="model_final",
9)Qwen/Qwen3.5-4B1DATALOADER:
2 ROBOVERSE:
3 cfg_opts: IMAGE.crop_img:0.9:IMAGE.img_size:224:IMAGE.cam_list:('3p1','wrist_right1')
4 cfg_path: libs/RoboVerse/roboverse/configs/ur5e_cluttered_pick_3obj_120.yaml
5 batch_size: 16
6 num_workers: 8
7EXP:
8 AMP: true
9 DATASET: roboverse
10 EXP_ID: lap_qwen3_5_4b_fft_ur5e_cluttered_pick_3obj_120_pyav
11 LOSS: {}
12 LR_SCHED: none
13 MODEL: qwen
14 OPTIMIZER: adamw
15 SEED: 0
16EXP_EXTRA:
17 no_test: true
18 no_track: true
19 no_val: true
20 save_at_steps:
21 - 4000
22 - 6000
23 - 8000
24 - 10000
25 - 12000
26 save_ckp: 0
27 save_last_ckpt: true
28 test_eval_freq: 1
29 val_eval_freq: 1
30LR_SCHED:
31 lr_clip: 1.0e-08
32 lr_decay_factor: 0.5
33 lr_patience: 4
34MODEL:
35 QWEN:
36 action_mask_aug_per: 0.4
37 action_type: original
38 add_vision_id: true
39 attention_dropout: 0.0
40 enable_thinking: true
41 grad_checkpoint: false
42 history: 1
43 horizon: 8
44 lap_action_is_absolute: true
45 lap_emit_holds: false
46 lap_rotation_precision: 1
47 lap_sum_decimal: 1f
48 lora_config: default
49 lora_rank: 8
50 num_bins_actions: 1000
51 num_cam: 2
52 original_action_dim: 7
53 qwen_model_id: Qwen/Qwen3.5-4B
54 reasoning: true
55 rgb_img_size:
56 - 224
57 - 224
58 rgb_input: true
59 tiled_rgb_imgs: true
60 use_flash_attention_2: true
61 use_lora: false
62 use_qlora: false
63TRAIN:
64 clip_grad_norm: 0.0
65 l2: 1.0e-10
66 lr: 1.0e-05
67 num_epochs: 100
68 num_iters: 12000
69 save_iter_ckp: 2500
70WANDB:
71 enable: true
72 entity: ''
73 log_interval: 100
74 mode: online
75 project: vla0
76 resume_id: ''
77 run_name: ''
78 tags: ''
79
80| File | Description |
|---|---|
model_final/model-*.safetensors | Full model weights |
model_final/config.json | Model configuration |
model_final/tokenizer.json | Tokenizer |
dataset_stats.pkl | Action normalization statistics (required for inference) |
config.yaml | Training configuration |