transformers
b61023a1 - 🚨🚨🚨 [eomt] make EoMT compatible with pipeline (#39122)

Commit
238 days ago
🚨🚨🚨 [eomt] make EoMT compatible with pipeline (#39122) * Make EoMT compatible with pipeline * Implicit patch offsets * remove patch offsets from arg * Modify tests * Update example * fix proc testcase * Add few more args * add pipeline test suite * fix * docstring fixes * add pipeline test * changes w.r.t review * 🙈 MB * should fix device mismatch * debug * Fixes device mismatch * use decorator * we can split mlp * expected values update --------- Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
Author
Parents
Loading