nvidia
/

nemotron-ocr-v1

object recognition

text recognition

layout analysis

Model card Files Files and versions

Fix Misaligned Address in batch mode

#9

by charlesb-nv - opened 23 days ago

base: refs/heads/main

←

from: refs/pr/9

Discussion Files changed

NVIDIA org 23 days ago

•

edited 23 days ago

Rebasing of fix from https://huggingface.co/nvidia/nemotron-ocr-v1/discussions/2

Fix misaligned address error in quad NMS cuda implementation4ba37061

charlesb-nv changed pull request status to open 23 days ago

Fix undefined idx variable in quad_non_maximal_suppression callThe for loop was removed but the [idx].unsqueeze(0) indexing wasleft behind, causing a NameError. Pass the full batch tensorsdirectly since the CUDA kernel now supports batch mode natively5b3a6420

emelryan changed pull request status to merged 22 days ago

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment