[Fix] Fix llama configs (open-compass#72)

Co-authored-by: Leymore <zfz-960727@163.com>
arajan10 · Jul 25, 2023 · 3715be6 · 3715be6
1 parent e9cdb24
commit 3715be6
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 12 deletions.
diff --git a/configs/models/hf_llama_13b.py b/configs/models/hf_llama_13b.py
@@ -1,13 +1,12 @@
 from opencompass.models import HuggingFaceCausalLM
 
-
 models = [
     # LLaMA 13B
     dict(
         type=HuggingFaceCausalLM,
         abbr='llama-13b-hf',
-        path="decapoda-research/llama-13b-hf",
-        tokenizer_path='decapoda-research/llama-13b-hf',
+        path="huggyllama/llama-13b",
+        tokenizer_path='huggyllama/llama-13b',
         tokenizer_kwargs=dict(padding_side='left',
                               truncation_side='left',
                               use_fast=False,

diff --git a/configs/models/hf_llama_30b.py b/configs/models/hf_llama_30b.py
@@ -1,13 +1,12 @@
 from opencompass.models import HuggingFaceCausalLM
 
-
 models = [
     # LLaMA 30B
     dict(
         type=HuggingFaceCausalLM,
         abbr='llama-30b-hf',
-        path="decapoda-research/llama-30b-hf",
-        tokenizer_path='decapoda-research/llama-30b-hf',
+        path="huggyllama/llama-30b",
+        tokenizer_path='huggyllama/llama-30b',
         tokenizer_kwargs=dict(padding_side='left',
                               truncation_side='left',
                               use_fast=False,

diff --git a/configs/models/hf_llama_65b.py b/configs/models/hf_llama_65b.py
@@ -1,13 +1,12 @@
 from opencompass.models import HuggingFaceCausalLM
 
-
 models = [
     # LLaMA 65B
     dict(
         type=HuggingFaceCausalLM,
         abbr='llama-65b-hf',
-        path="decapoda-research/llama-65b-hf",
-        tokenizer_path='decapoda-research/llama-65b-hf',
+        path="huggyllama/llama-65b",
+        tokenizer_path='huggyllama/llama-65b',
         tokenizer_kwargs=dict(padding_side='left',
                               truncation_side='left',
                               use_fast=False,

diff --git a/configs/models/hf_llama_7b.py b/configs/models/hf_llama_7b.py
@@ -1,13 +1,12 @@
 from opencompass.models import HuggingFaceCausalLM
 
-
 models = [
     # LLaMA 7B
     dict(
         type=HuggingFaceCausalLM,
         abbr='llama-7b-hf',
-        path="decapoda-research/llama-7b-hf",
-        tokenizer_path='decapoda-research/llama-7b-hf',
+        path="huggyllama/llama-7b",
+        tokenizer_path='huggyllama/llama-7b',
         tokenizer_kwargs=dict(padding_side='left',
                               truncation_side='left',
                               use_fast=False,