fix bugs

transfer.py

@@ -3,10 +3,19 @@ import os
 import glob
 import numpy as np
 
+def convert_model_name(model):
+    name = {
+        "T5": "google-flan-t5-large",
+        "UL2": "google-flan-ul2",
+        "Vicuna": "vicuna-13b",
+        "ChatGPT": "chatgpt",
+    }
+    return name[model]
 
 def retrieve_transfer(source, target, attack, shot):
-    source = source.replace("/", "-")
-    target = target.replace("/", "-")
+    source = convert_model_name(source)
+    target = convert_model_name(target)
+    attack = attack.lower()
     file_dir = "./results_transfer/"+source+"_"+target+"/"+attack+"_"+str(shot)+"_shot.json"
     with open(file_dir, 'r', encoding='utf-8') as f:
         data = json.load(f)