From bc573b22d1b68226d329e43f7a882677e1d3cb51 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?R=C3=A9mi=20Louf?= <remilouf@gmail.com>
Date: Thu, 7 Dec 2023 09:40:20 +0100
Subject: [PATCH] Fix the width of the code example on index

---
 docs/index.md              | 2 +-
 docs/stylesheets/extra.css | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/docs/index.md b/docs/index.md
index c70f11e72..73e598af8 100644
--- a/docs/index.md
+++ b/docs/index.md
@@ -40,7 +40,7 @@ class Character(BaseModel):
     strength: int
 
 
-model = models.transformers("mistralai/Mistral-7B-v0.1", device="cuda")
+model = models.transformers("mistralai/Mistral-7B-v0.1")
 generator = generate.json(model, Character, max_tokens=100)
 sequence = generator("Give me a character description")
 ```
diff --git a/docs/stylesheets/extra.css b/docs/stylesheets/extra.css
index d98248e9b..51420e3b6 100644
--- a/docs/stylesheets/extra.css
+++ b/docs/stylesheets/extra.css
@@ -14,7 +14,7 @@ code > * {
 
 .index-pre-code {
   margin: 50px;
-  width: 60%;
+  max-width: 700px;
   left: 50%;
 }