Skip to content

Commit e7969a3

Browse files
rsyuecirquit
andauthored
Update src/fairseq2/models/hg/adapter.py
Co-authored-by: Alexander Erben <aerben@meta.com>
1 parent 9048cf4 commit e7969a3

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/fairseq2/models/hg/adapter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -240,7 +240,7 @@ def forward( # type: ignore[override]
240240
else:
241241
return loss
242242
else:
243-
# Inference mode - just get logits
243+
# Inference mode returns logits only
244244
outputs = self._hf_model(**hf_kwargs)
245245
return cast(Tensor, outputs.logits)
246246

0 commit comments

Comments
 (0)