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 Feb 5

base: refs/heads/main

←

from: refs/pr/9

Discussion Files changed

NVIDIA org Feb 5

•

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 Feb 5

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 Feb 6

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