fix int/str for conv_dim indexing

#9

Implementing the same fix by @winglian for hymba-base: https://huggingface.co/nvidia/Hymba-1.5B-Base/discussions/5

for this hymba-instruct model:

SimonX changed pull request status to merged
NVIDIA org

Thanks! Merged

Sign up or log in to comment