Spaces:
Runtime error
Runtime error
File size: 617 Bytes
683d0f6 8bab3c8 683d0f6 8bab3c8 683d0f6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
import json
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 = 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)
return data
|