Prompt-Transferability

1
import torch
2
import os
3

4
dir = os.listdir()
5
for l in dir:
6
    if ".py" in l:
7
        continue
8
    #a = torch.load(l+"/task_prompt").type(torch.float32)
9
    a = torch.load(l+"/latest.pt").type(torch.float32)
10
    print(a)
11
    if int(a.shape[0]) == 100 and int(a.shape[1]) == 4096:
12
        print(l+"/task_prompt", "Done")
13
        continue
14
    else:
15
        a = a.reshape(int(a.shape[0])*int(a.shape[1]),int(a.shape[2]))
16
    torch.save(a,l+"/task_prompt")
17
    print(l+"/task_prompt", "Done")
18

19