Add files via upload

Author: travis-xu

run_experiment.py

@@ -4,7 +4,7 @@
 # from sentence_transformers import SentenceTransformer, util
 import json
 import re
-from utils_xqc import standardize, standardize_category, change_name, set_seed, load_json, save_json
+from utils import set_seed, load_json, save_json
 import random
 import numpy as np
 import json

utils.py

@@ -0,0 +1,26 @@
+import json
+import re
+import torch
+import transformers
+import transformers.models.llama.modeling_llama
+from functools import partial
+import numpy as np
+import random
+
+def set_seed(seed):
+    torch.manual_seed(seed)
+    torch.cuda.manual_seed_all(seed)
+    np.random.seed(seed)
+    random.seed(seed)
+    # torch.backends.cudnn.deterministic = True
+
+
+def load_json(file_name, file_path=""):
+    file = []
+    with open(file_path + file_name, "r") as f:
+        file = json.load(f)
+    return file
+
+def save_json(data, file_name, file_path=""):
+    with open(file_path + file_name, "w") as f:
+        json.dump(data, f)