Add NanotronLightevalModel to Model docs

huggingface · Dec 3, 2024 · 5ab3262 · 5ab3262
1 parent 7b38ead
commit 5ab3262
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/docs/source/package_reference/models.mdx b/docs/source/package_reference/models.mdx
@@ -7,7 +7,6 @@
 ## Accelerate and Transformers Models
 ### BaseModel
 [[autodoc]] models.base_model.BaseModel
-[//]: # (TODO: Fix import error)
 ### AdapterModel
 [[autodoc]] models.adapter_model.AdapterModel
 ### DeltaModel
@@ -19,10 +18,9 @@
 ### ModelClient
 [[autodoc]] models.tgi_model.ModelClient
 
-[//]: # (TODO: Fix import error)
-[//]: # (## Nanotron Model)
-[//]: # (### NanotronLightevalModel)
-[//]: # ([[autodoc]] models.nanotron_model.NanotronLightevalModel)
+## Nanotron Model
+### NanotronLightevalModel
+[[autodoc]] models.nanotron_model.NanotronLightevalModel
 
 [//]: # (TODO: Fix import error)
 [//]: # (## VLLM Model)