Spaces:
Runtime error
Runtime error
Commit
Β·
89bf924
1
Parent(s):
c5c065e
Debug
Browse files
app.py
CHANGED
@@ -10,8 +10,8 @@ class Interactive:
|
|
10 |
self.tokenizer = transformers.AutoTokenizer.from_pretrained('liujch1998/cd-pi', use_auth_token=os.environ['HF_TOKEN_DOWNLOAD'])
|
11 |
self.model = transformers.AutoModelForSeq2SeqLM.from_pretrained('liujch1998/cd-pi', use_auth_token=os.environ['HF_TOKEN_DOWNLOAD']).to(device)
|
12 |
self.linear = torch.nn.Linear(self.model.shared.embedding_dim, 1).to(device)
|
13 |
-
self.linear.weight = torch.nn.Parameter(self.model.shared.weight[32099, :].unsqueeze(0)) # (1, D)
|
14 |
-
self.linear.bias = torch.nn.Parameter(self.model.shared.weight[32098, 0].unsqueeze(0)) # (1)
|
15 |
self.model.eval()
|
16 |
self.t = 2.2247
|
17 |
|
|
|
10 |
self.tokenizer = transformers.AutoTokenizer.from_pretrained('liujch1998/cd-pi', use_auth_token=os.environ['HF_TOKEN_DOWNLOAD'])
|
11 |
self.model = transformers.AutoModelForSeq2SeqLM.from_pretrained('liujch1998/cd-pi', use_auth_token=os.environ['HF_TOKEN_DOWNLOAD']).to(device)
|
12 |
self.linear = torch.nn.Linear(self.model.shared.embedding_dim, 1).to(device)
|
13 |
+
# self.linear.weight = torch.nn.Parameter(self.model.shared.weight[32099, :].unsqueeze(0)) # (1, D)
|
14 |
+
# self.linear.bias = torch.nn.Parameter(self.model.shared.weight[32098, 0].unsqueeze(0)) # (1)
|
15 |
self.model.eval()
|
16 |
self.t = 2.2247
|
17 |
|