Searched refs:LlamaModel (Results 1 – 5 of 5) sorted by relevance
37 LlamaModel,251 llama_instance = LlamaModel(config, output_new_cache_only=True)
30 LlamaModel,351 llama_instance = LlamaModel(config, output_new_cache_only=True)
588 from transformers.models.llama.modeling_llama import LlamaModel # noqa: F811606 self.llama = LlamaModel(config)
218 class LlamaModel(nn.Module): class
494 model = transformers.LlamaModel(config).eval()