Update HuggingFace transformers support

README.md

@@ -18,6 +18,7 @@ library_name: transformers
 <img src="assets/EXAONE_Symbol+BI_3d.png", width="300", style="margin: 40 auto;">
 🎉 License Updated! We are pleased to announce our more flexible licensing terms 🤗
 <br>✈️ Try on <a href="https://friendli.ai/suite/~/serverless-endpoints/LGAI-EXAONE/EXAONE-4.0-32B/overview">FriendliAI</a>
+<br><br><i>📢 EXAONE 4.0 is officially supported by HuggingFace transformers! Please check out the guide <a href="#quickstart">below</a></i>
 <br>
 
 # EXAONE-4.0-1.2B
@@ -48,14 +49,7 @@ For more details, please refer to our [technical report](https://arxiv.org/abs/2
 
 ## Quickstart
 
-You should install the transformers library forked from the original, available in our [PR](https://github.com/huggingface/transformers/pull/39129).
-Once this PR is merged and released, we will update this section. 
-
-You can install the latest version of transformers with support for EXAONE 4.0 by following the command:
-
-```bash
-pip install git+https://github.com/lgai-exaone/transformers@add-exaone4
-```
+You should install the transformers library with version >= `4.54.0`.
 
 ### Non-reasoning mode
 

config.json

@@ -10,6 +10,38 @@
   "hidden_size": 2048,
   "initializer_range": 0.02,
   "intermediate_size": 4096,
+  "layer_types": [
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention"
+  ],
   "max_position_embeddings": 65536,
   "model_type": "exaone4",
   "num_attention_heads": 32,
@@ -29,7 +61,7 @@
   "sliding_window_pattern": null,
   "tie_word_embeddings": true,
   "torch_dtype": "bfloat16",
-  "transformers_version": "4.54.0.dev0",
+  "transformers_version": "4.54.0",
   "use_cache": false,
   "vocab_size": 102400
 }

generation_config.json

@@ -3,5 +3,5 @@
   "bos_token_id": 1,
   "eos_token_id": 361,
   "pad_token_id": 0,
-  "transformers_version": "4.54.0.dev0"
+  "transformers_version": "4.54.0"
 }