Update

huggingface · Feb 14, 2024 · ce168e1 · ce168e1
1 parent d31f742
commit ce168e1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/examples/pytorch/test_pytorch_examples.py b/examples/pytorch/test_pytorch_examples.py
@@ -159,7 +159,7 @@ def test_run_clm_config_overrides(self):
         tmp_dir = self.get_auto_remove_tmp_dir()
         testargs = f"""
             run_clm.py
-            --model_type openai-community/gpt2
+            --model_type gpt2
             --tokenizer_name openai-community/gpt2
             --train_file ./tests/fixtures/sample_text.txt
             --output_dir {tmp_dir}
@@ -315,7 +315,7 @@ def test_generation(self):
             testargs.append("--fp16")
 
         model_type, model_name = (
-            "--model_type=openai-community/gpt2",
+            "--model_type=gpt2",
             "--model_name_or_path=sshleifer/tiny-gpt2",
         )
         with patch.object(sys, "argv", testargs + [model_type, model_name]):

diff --git a/src/transformers/models/openai/configuration_openai.py b/src/transformers/models/openai/configuration_openai.py
@@ -110,7 +110,7 @@ class OpenAIGPTConfig(PretrainedConfig):
     >>> configuration = model.config
     ```"""
 
-    model_type = "openai-community/openai-gpt"
+    model_type = "openai-gpt"
     attribute_map = {
         "max_position_embeddings": "n_positions",
         "hidden_size": "n_embd",