Views
No views yet
/remote-home/share/chenglong/Workplace/SpatialVLA/ckpts_pretrained/spatialvla-4b-224-pt/remote-home/share/chenglong/Workplace/SpatialVLA/outputs/spatialvla_4b_finetune/2025-10-21/03-55-14_glasses_sigma5_dataset_spatialvla-4b-224-pt_lr5e-6_bs32_node1_gpu2_r32_a32_ep10_linear+emb+h/checkpoint-300001import torch
2from transformers import AutoModel, AutoProcessor
3
4model_path = "/remote-home/share/chenglong/Workplace/SpatialVLA/ckpts_merged/glassvla-4b-sam2-lora-percent10-30k-sigma-5"
5processor = AutoProcessor.from_pretrained(model_path, trust_remote_code=True)
6model = AutoModel.from_pretrained(
7 model_path,
8 trust_remote_code=True,
9 torch_dtype=torch.bfloat16
10).eval().cuda()
11
12# Use the model for inference
13# ... your inference code here ...