edited functions

run_llm.py

@@ -74,13 +74,13 @@ for m in model_mapping.keys():
 
 s = 0
 e = 1000
-with open('test/sample_uniform_1k_2.txt', 'r') as f:
+with open('sample_uniform_1k_2.txt', 'r') as f:
     selected_idx = f.readlines()
 selected_idx = [int(i.strip()) for i in selected_idx][s:e]
 
 
 ptb = []
-with open('test/ptb.jsonl', 'r') as f:
+with open('ptb.jsonl', 'r') as f:
     for l in f:
         ptb.append(json.loads(l))