Upload Isoformer
Browse files- modeling_isoformer.py +2 -7
- pytorch_model.bin +1 -1
modeling_isoformer.py
CHANGED
|
@@ -113,7 +113,6 @@ class Isoformer(PreTrainedModel):
|
|
| 113 |
tensor_dna,
|
| 114 |
tensor_rna,
|
| 115 |
tensor_protein,
|
| 116 |
-
attention_mask_dna,
|
| 117 |
attention_mask_rna,
|
| 118 |
attention_mask_protein
|
| 119 |
):
|
|
@@ -159,10 +158,6 @@ class Isoformer(PreTrainedModel):
|
|
| 159 |
|
| 160 |
|
| 161 |
return {
|
| 162 |
-
"gene_expression_predictions":x,
|
| 163 |
-
"
|
| 164 |
-
"final_embeddings": final_dna_embeddings,
|
| 165 |
-
"dna_embedding": dna_embedding,
|
| 166 |
-
"rna_embedding": rna_embedding,
|
| 167 |
-
"protein_embedding": protein_embedding
|
| 168 |
}
|
|
|
|
| 113 |
tensor_dna,
|
| 114 |
tensor_rna,
|
| 115 |
tensor_protein,
|
|
|
|
| 116 |
attention_mask_rna,
|
| 117 |
attention_mask_protein
|
| 118 |
):
|
|
|
|
| 158 |
|
| 159 |
|
| 160 |
return {
|
| 161 |
+
"gene_expression_predictions": x,
|
| 162 |
+
"final_dna_embeddings": final_dna_embeddings,
|
|
|
|
|
|
|
|
|
|
|
|
|
| 163 |
}
|
pytorch_model.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 2803153818
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d8fefc8c13856b1b9109f34fed3974e69186a296931c1e230e26f6aca8f268bb
|
| 3 |
size 2803153818
|