Test

Author: regisss

docs/combine_docs.py

@@ -106,23 +106,32 @@ def add_subpackage_doc(base_toc: List, subpackage: str):
         subpackage (str): subpackage name
     """
 
-    section_title = EXTERNAL_DOCUMENTATION[subpackage]
-    # Update optimum table of contents
-    base_toc.insert(
-        SUBPACKAGE_TOC_INSERT_INDEX,
-        {
-            "sections": [
-                {
-                    # Ideally this should directly point at https://huggingface.co/docs/optimum-xxx/index
-                    # Current hacky solution is to have a redirection in _redirects.yml
-                    "local": f"docs/optimum-{subpackage}/index",
-                    "title": f"🤗 Optimum {section_title}",
-                }
-            ],
-            "title": TITLE_SECTION.get(section_title, section_title),
-            "isExpanded": False,
-        },
-    )
+    if subpackage == "habana":
+        new_section = {
+            "local": f"docs/optimum-habana/index",
+            "title": f"🤗 Optimum For Intel Gaudi",
+        }
+        for i, toc_element in enumerate(base_toc):
+            if toc_element["title"] == "Intel":
+                base_toc[i]["sections"].append(new_section)
+    else:
+        section_title = EXTERNAL_DOCUMENTATION[subpackage]
+        # Update optimum table of contents
+        base_toc.insert(
+            SUBPACKAGE_TOC_INSERT_INDEX,
+            {
+                "sections": [
+                    {
+                        # Ideally this should directly point at https://huggingface.co/docs/optimum-xxx/index
+                        # Current hacky solution is to have a redirection in _redirects.yml
+                        "local": f"docs/optimum-{subpackage}/index",
+                        "title": f"🤗 Optimum {section_title}",
+                    }
+                ],
+                "title": TITLE_SECTION.get(section_title, section_title),
+                "isExpanded": False,
+            },
+        )
 
 
 def main():

docs/source/index.mdx

@@ -47,8 +47,8 @@ The packages below enable you to get the best of the 🤗 Hugging Face ecosystem
       <p class="text-gray-700">Accelerate your training and inference workflows with <span class="underline" onclick="event.preventDefault(); window.open('https://cloud.google.com/tpu', '_blank');">Google TPUs</span></p>
     </a>
     <a class="!no-underline border dark:border-gray-700 p-5 rounded-lg shadow hover:shadow-lg" href="https://huggingface.co/docs/optimum-habana/index"
-      ><div class="w-full text-center bg-gradient-to-br from-indigo-400 to-indigo-500 rounded-lg py-1.5 font-semibold mb-5 text-white text-lg leading-relaxed">Habana</div>
-      <p class="text-gray-700">Maximize training throughput and efficiency with <span class="underline" onclick="event.preventDefault(); window.open('https://docs.habana.ai/en/latest/Gaudi_Overview/Gaudi_Architecture.html', '_blank');">Habana's Gaudi processor</span></p>
+      ><div class="w-full text-center bg-gradient-to-br from-indigo-400 to-indigo-500 rounded-lg py-1.5 font-semibold mb-5 text-white text-lg leading-relaxed">Intel Gaudi</div>
+      <p class="text-gray-700">Maximize training throughput and efficiency with <span class="underline" onclick="event.preventDefault(); window.open('https://docs.habana.ai/en/latest/Gaudi_Overview/Gaudi_Architecture.html', '_blank');">Intel's Gaudi processor</span></p>
     </a>
     <a class="!no-underline border dark:border-gray-700 p-5 rounded-lg shadow hover:shadow-lg" href="./furiosa/index"
       ><div class="w-full text-center bg-gradient-to-br from-green-400 to-green-500 rounded-lg py-1.5 font-semibold mb-5 text-white text-lg leading-relaxed">FuriosaAI</div>