Shashank Verma

icon indicating an email [email protected]

icon company Nvidia icon location Deep Learning Developer Advocate @ Nvidia

Results 2 issues of Shashank Verma

Minor fix in gemma 2 ckpt conversion to support Gemma 2B

# What does this PR do ? Add a one line overview of what this PR aims to accomplish. **Collection**: [Note which collection this PR will affect] # Changelog -...

NLP

Fix missing runtime_gather_output arg in gemma 2 output layer

# What does this PR do ? * The forward pass in Gemma 2 passes runtime_gather_output * Leads to unexpected keyword arg error while finetuning gemma 2 2b without it...

NLP