From 6353a87f4c3b9919d31eeda735187a1f9fc72c04 Mon Sep 17 00:00:00 2001
From: fxmarty <9808326+fxmarty@users.noreply.github.com>
Date: Wed, 8 Nov 2023 19:09:30 +0900
Subject: [PATCH] Update
 server/text_generation_server/models/custom_modeling/idefics_modeling.py

---
 .../models/custom_modeling/idefics_modeling.py                  | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/server/text_generation_server/models/custom_modeling/idefics_modeling.py b/server/text_generation_server/models/custom_modeling/idefics_modeling.py
index 4e96ec81..9c3bfa4f 100644
--- a/server/text_generation_server/models/custom_modeling/idefics_modeling.py
+++ b/server/text_generation_server/models/custom_modeling/idefics_modeling.py
@@ -432,6 +432,8 @@ class IdeficsRMSNorm(nn.Module):
                 out = out.view(*shape)
 
             return out
+        else:
+            raise ValueError("Your system seem to be not supported. Please check your install or open an issue at https://github.com/huggingface/text-generation-inference/issues with a clear reproduction.")
 
 
 # this was adapted from LlamaMLP