Refinements

huggingface · Jul 21, 2023 · a23ddcd · a23ddcd
1 parent 83e9f6c
commit a23ddcd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git 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
@@ -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"