Update modeling_custom.py
Browse files- modeling_custom.py +0 -1
modeling_custom.py
CHANGED
|
@@ -148,7 +148,6 @@ class Gemma2ForQuantileSequenceClassification(Gemma2PreTrainedModel):
|
|
| 148 |
`config.num_labels > 1` a classification loss is computed (Cross-Entropy).
|
| 149 |
"""
|
| 150 |
return_dict = return_dict if return_dict is not None else self.config.use_return_dict
|
| 151 |
-
print(input_ids)
|
| 152 |
transformer_outputs = self.model(
|
| 153 |
input_ids,
|
| 154 |
attention_mask=attention_mask,
|
|
|
|
| 148 |
`config.num_labels > 1` a classification loss is computed (Cross-Entropy).
|
| 149 |
"""
|
| 150 |
return_dict = return_dict if return_dict is not None else self.config.use_return_dict
|
|
|
|
| 151 |
transformer_outputs = self.model(
|
| 152 |
input_ids,
|
| 153 |
attention_mask=attention_mask,
|