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-27/09-34-46_glasses_sigma12_dataset_spatialvla-4b-224-pt_lr5e-6_bs16_node1_gpu2_r32_a32_ep10_linear+emb+h/checkpoint-634901import torch
2from transformers import AutoModel, AutoProcessor
3
4model_path = "/remote-home/share/chenglong/Workplace/SpatialVLA/ckpts_merged/glassvla-4b-sam2-lora-percent10-63k-sigma-12"
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 ...