Update README.md

README.md

@@ -6,9 +6,10 @@ base_model:
 nvidia/Alpamayo-R1-10B 4bit Model. 
 
 
-----  model download  ./Alpamayo-R1-10B-4bit 
-----  GPU 12G Memory Run abble, num_frames is 1 ~ 8, over OOM..
+ model download  ./Alpamayo-R1-10B-4bit 
+ GPU 12G Memory Run abble, num_frames is 1 ~ 8, over OOM
 
+-------------------------
 import torch
 import numpy as np
 
@@ -16,12 +17,12 @@ from alpamayo_r1.models.alpamayo_r1 import AlpamayoR1
 from alpamayo_r1.load_physical_aiavdataset import load_physical_aiavdataset
 from alpamayo_r1 import helper
 
-# Example clip ID
+ 
 model_path = "Alpamayo-R1-10B-4bit"
 model = AlpamayoR1.from_pretrained(model_path, dtype=torch.bfloat16).to("cuda")
 processor = helper.get_processor(model.tokenizer)
 
-# Example clip ID
+ 
 clip_id = "030c760c-ae38-49aa-9ad8-f5650a545d26"
 print(f"Loading dataset for clip_id: {clip_id}...")
 data = load_physical_aiavdataset(clip_id, t0_us=15_100_000,num_frames=1)
@@ -57,7 +58,7 @@ with torch.autocast("cuda", dtype=torch.bfloat16):
         return_extra=True,
     )
 
-# the size is [batch_size, num_traj_sets, num_traj_samples]
+ 
 print("Chain-of-Causation (per trajectory):\n", extra["cot"][0])
 
 gt_xy = data["ego_future_xyz"].cpu()[0, 0, :, :2].T.numpy()