From 4217ddb842ddd3c8e2e1f212792f01b4a2cdcd82 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Tue, 9 Apr 2024 17:19:26 +0000
Subject: [PATCH] Move import up.

---
 server/text_generation_server/models/vlm_causal_lm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/server/text_generation_server/models/vlm_causal_lm.py b/server/text_generation_server/models/vlm_causal_lm.py
index 8a647a09..16042fc9 100644
--- a/server/text_generation_server/models/vlm_causal_lm.py
+++ b/server/text_generation_server/models/vlm_causal_lm.py
@@ -1,5 +1,6 @@
 import re
 import torch
+import math
 from PIL import Image
 from io import BytesIO
 import base64
@@ -80,7 +81,6 @@ def get_number_of_features(height: int, width: int, config) -> int:
         image_grid_pinpoints,
         image_size,
     )
-    import math
 
     height_of_patch = math.ceil(height / width * npatches)