Update chatNT.py
Browse files
chatNT.py
CHANGED
|
@@ -733,6 +733,7 @@ class TorchMultiOmicsModel(PreTrainedModel):
|
|
| 733 |
)
|
| 734 |
|
| 735 |
outs["logits"] = logits
|
|
|
|
| 736 |
|
| 737 |
return outs
|
| 738 |
|
|
|
|
| 733 |
)
|
| 734 |
|
| 735 |
outs["logits"] = logits
|
| 736 |
+
outs["projected_bio_embeddings_end"] = projected_bio_embeddings.clone()
|
| 737 |
|
| 738 |
return outs
|
| 739 |
|