Testing

run_llm.py

@@ -94,7 +94,7 @@ selected_idx = [int(i.strip()) for i in selected_idx]#[s:e]
 
 
 ptb = []
-with open('sample_uniform_1k_2.txt', 'r') as f:
+with open('ptb.jsonl', 'r') as f:
     for l in f:
         ptb.append(json.loads(l))