Skip to content

Commit 81a04a1

Browse files
Fix UnboundLocalError when auxiliary_loss is False
Initialize auxiliary_outputs to None before conditional block to prevent UnboundLocalError when config.auxiliary_loss is False. The function always returns auxiliary_outputs, but it was only initialized inside the if config.auxiliary_loss block, causing an error when the condition is False. Fixes #42219
1 parent 66d5711 commit 81a04a1

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/transformers/loss/loss_rt_detr.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -447,6 +447,7 @@ def RTDetrForObjectDetectionLoss(
447447
outputs_loss = {}
448448
outputs_loss["logits"] = logits
449449
outputs_loss["pred_boxes"] = pred_boxes
450+
auxiliary_outputs = None
450451
if config.auxiliary_loss:
451452
if denoising_meta_values is not None:
452453
dn_out_coord, outputs_coord = torch.split(outputs_coord, denoising_meta_values["dn_num_split"], dim=2)

0 commit comments

Comments
 (0)