Error in NXD LLama Inference

[EmilyWebber]

Hi there, looks like we have a uncompleted function definition on this line. Can you take a look?

neuronx-distributed/examples/inference/llama2/neuron_modeling_llama.py

Line 297 in 4f95471

def init_model(self, config: NeuronLlamaConfig):