Repository navigation

Refactor GPT-Neo output tracing to use capture_outputs/can_return_tuple - #44018

Closed

yashbora9 wants to merge 2 commits into

huggingface:mainfrom

yashbora9:refactor-gpt-neo-output-tracing

Closed

Refactor GPT-Neo output tracing to use capture_outputs/can_return_tuple#44018
yashbora9 wants to merge 2 commits into
huggingface:mainfrom
yashbora9:refactor-gpt-neo-output-tracing

Commits on Feb 15, 2026
- Refactor GPT-Neo to use @capture_outputs and @can_return_tuple decorators
  
  authored and
  Yash Bora
  committed
  73b2f36
Commits on Feb 16, 2026
- Add @merge_with_config_defaults and fix test for output_attentions removal
  
  authored and
  Yash Bora
  committed
  0df1861