From a23ddcd5e3210b9438f10791e57b5456f00a6190 Mon Sep 17 00:00:00 2001
From: regisss <regis@huggingface.co>
Date: Fri, 21 Jul 2023 09:34:54 +0200
Subject: [PATCH] Refinements

---
 docs/combine_docs.py  | 5 ++++-
 docs/source/index.mdx | 2 +-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/docs/combine_docs.py b/docs/combine_docs.py
index 8715f1a8f4..7ab04749da 100755
--- a/docs/combine_docs.py
+++ b/docs/combine_docs.py
@@ -35,7 +35,10 @@ def rename_subpackage_toc(subpackage: str, toc: Dict):
                 rename_subpackage_toc(subpackage, [file])
 
     # Just keep the name of the partner
-    toc[0]["title"] = toc[0]["title"].split("Optimum ")[-1]
+    if subpackage != "furiosa":
+        toc[0]["title"] = toc[0]["title"].split("Optimum ")[-1]
+    else:
+        toc[0]["title"] = "FuriosaAI"
 
 
 def rename_copy_subpackage_html_paths(subpackage: str, subpackage_path: Path, optimum_path: Path, version: str):
diff --git a/docs/source/index.mdx b/docs/source/index.mdx
index cfd6c98464..04f2af3aa9 100644
--- a/docs/source/index.mdx
+++ b/docs/source/index.mdx
@@ -34,7 +34,7 @@ As such, Optimum enables developers to efficiently use any of these platforms wi
       <p class="text-gray-700">Accelerate your training and inference workflows with <a href="https://aws.amazon.com/machine-learning/trainium/">AWS Trainium</a> and <a href="https://aws.amazon.com/machine-learning/inferentia/">AWS Inferentia</a></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">Furiosa</div>
+      ><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>
       <p class="text-gray-700">Fast and efficient inference on <a href="https://www.furiosa.ai/">FuriosaAI WARBOY</a></p>
     </a>
     <a class="!no-underline border dark:border-gray-700 p-5 rounded-lg shadow hover:shadow-lg" href="./onnxruntime/overview"