Upload model
Browse files- modeling_phylogpn.py +2 -1
modeling_phylogpn.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
from typing import List, Optional
|
2 |
|
3 |
import torch
|
@@ -235,7 +236,7 @@ class PhyloGPNModel(PreTrainedModel):
|
|
235 |
kernel_size=config.kernel_size,
|
236 |
pad_token_idx=5
|
237 |
)
|
238 |
-
|
239 |
def get_embeddings(self, input_ids: torch.Tensor):
|
240 |
return self._model.get_embeddings(input_ids)
|
241 |
|
|
|
1 |
+
from functools import cached_property
|
2 |
from typing import List, Optional
|
3 |
|
4 |
import torch
|
|
|
236 |
kernel_size=config.kernel_size,
|
237 |
pad_token_idx=5
|
238 |
)
|
239 |
+
|
240 |
def get_embeddings(self, input_ids: torch.Tensor):
|
241 |
return self._model.get_embeddings(input_ids)
|
242 |
|