Update modeling_gemma_3_omni.py
Browse files- modeling_gemma_3_omni.py +2 -1
modeling_gemma_3_omni.py
CHANGED
|
@@ -373,7 +373,8 @@ class Gemma3OmniForConditionalGeneration(Gemma3PreTrainedModel, GenerationMixin)
|
|
| 373 |
)
|
| 374 |
|
| 375 |
hidden_states = outputs[0]
|
| 376 |
-
|
|
|
|
| 377 |
logits = None
|
| 378 |
|
| 379 |
if skip_logits and labels is None:
|
|
|
|
| 373 |
)
|
| 374 |
|
| 375 |
hidden_states = outputs[0]
|
| 376 |
+
|
| 377 |
+
loss = None
|
| 378 |
logits = None
|
| 379 |
|
| 380 |
if skip_logits and labels is None:
|