riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Fix Base Model Name of LlamaForQuestionAnswering (#29258)

Browse Source 
* LlamaForQuestionAnswering self.transformer->self.model

* fix "Copied from" string

* Llama QA model: set base_model_prefix = "transformer"
This commit is contained in:
Leon Engländer 2024-03-01 02:58:19 +01:00 committed by GitHub
parent 5ee0868a4b
commit 2858d6c634
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/llama/modeling_llama.py
View File

@ -1454,6 +1454,8 @@ SQuAD (a linear layer on top of the hidden-states output to compute `span start
LLAMA_START_DOCSTRING, LLAMA_START_DOCSTRING,
) )
class LlamaForQuestionAnswering(LlamaPreTrainedModel): class LlamaForQuestionAnswering(LlamaPreTrainedModel):
base_model_prefix = "transformer"
# Copied from transformers.models.bloom.modeling_bloom.BloomForQuestionAnswering.__init__ with Bloom->Llama # Copied from transformers.models.bloom.modeling_bloom.BloomForQuestionAnswering.__init__ with Bloom->Llama
def __init__(self, config): def __init__(self, config):
super().__init__(config) super().__init__(config)