Searched refs:LlamaDecoderLayer (Results 1 – 4 of 4) sorted by relevance
41 class LlamaDecoderLayer(DecoderLayer): class76 decoder_class=LlamaDecoderLayer,
496 LlamaDecoderLayer,517 self.decoder = LlamaDecoderLayer(config, layer_idx=0)520 self.decoder = LlamaDecoderLayer(config)
185 class LlamaDecoderLayer(nn.Module): class234 LlamaDecoderLayer(config, self.output_new_cache_only)
2348 from transformers.models.llama.modeling_llama import LlamaDecoderLayer2365 "llama_v2_7b_16h": (LlamaDecoderLayer,),