Update geneformer/perturber_utils.py
Browse files
geneformer/perturber_utils.py
CHANGED
|
@@ -161,10 +161,6 @@ def get_model_emb_dims(model):
|
|
| 161 |
def get_model_input_size(model):
|
| 162 |
return model.config.max_position_embeddings
|
| 163 |
|
| 164 |
-
def get_model_input_size(model):
|
| 165 |
-
return int(re.split("\(|,", str(model.bert.embeddings.position_embeddings))[1])
|
| 166 |
-
|
| 167 |
-
|
| 168 |
def flatten_list(megalist):
|
| 169 |
return [item for sublist in megalist for item in sublist]
|
| 170 |
|
|
|
|
| 161 |
def get_model_input_size(model):
|
| 162 |
return model.config.max_position_embeddings
|
| 163 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 164 |
def flatten_list(megalist):
|
| 165 |
return [item for sublist in megalist for item in sublist]
|
| 166 |
|