fix: count only attention layers in NemotronH KV memory - #544

fix: count only attention layers in NemotronH KV memory#544

milesial merged 1 commit into

ai-dynamo:mainai-dynamo/aisimulate:mainfrom

milesial:alexandrem/fix-nemotron-h-attention-kvmilesial/aisimulate-public:alexandrem/fix-nemotron-h-attention-kvCopy head branch name to clipboard