calbors commited on
Commit
a723648
·
verified ·
1 Parent(s): 4dae406

Upload model

Browse files
Files changed (1) hide show
  1. 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