Modeling fixed
Browse files- modeling_nicheformer.py +0 -1
modeling_nicheformer.py
CHANGED
|
@@ -124,7 +124,6 @@ class NicheformerModel(NicheformerPreTrainedModel):
|
|
| 124 |
else:
|
| 125 |
embeddings = self.positional_embedding(token_embedding)
|
| 126 |
|
| 127 |
-
print("embeddings NicheformerModel: ", embeddings)
|
| 128 |
# Process through transformer layers up to desired layer
|
| 129 |
if layer < 0:
|
| 130 |
layer = self.config.nlayers + layer # -1 means last layer
|
|
|
|
| 124 |
else:
|
| 125 |
embeddings = self.positional_embedding(token_embedding)
|
| 126 |
|
|
|
|
| 127 |
# Process through transformer layers up to desired layer
|
| 128 |
if layer < 0:
|
| 129 |
layer = self.config.nlayers + layer # -1 means last layer
|