transformers
9556054f - OWL-ViT box_predictor inefficiency issue (#29712)

Commit
1 year ago
OWL-ViT box_predictor inefficiency issue (#29712) * Calculating box_bias at the start once, then reusing it at inference * Updating the compute_box_bias function for backwards compatibility * Caching compute_box_bias function * Bux fix * Update owlv2 accordingly to ensure repo consistency * Co-authored by: nvbinh15 <binh.pdc01@gmail.com> * Fixup changes * Made copied code consistent * Co-authored by: nvbinh15 <binh.pdc01@gmail.com> --------- Co-authored-by: Nguyen Van Binh <> Co-authored-by: Nguyen Van Binh <binh.pdc01@gmail.com>
Author
Parents
Loading