From 549df839d78a8294a6dbab12df4f31b618590887 Mon Sep 17 00:00:00 2001 From: ssmi153 <129111316+ssmi153@users.noreply.github.com> Date: Wed, 12 Jul 2023 23:26:09 +0800 Subject: [PATCH] Tuple rather than list of exception types --- server/text_generation_server/utils/weights.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/server/text_generation_server/utils/weights.py b/server/text_generation_server/utils/weights.py index d254aa52..723d0558 100644 --- a/server/text_generation_server/utils/weights.py +++ b/server/text_generation_server/utils/weights.py @@ -130,7 +130,7 @@ class Weights: try: bits = self.get_tensor("gptq_bits").item() groupsize = self.get_tensor("gptq_groupsize").item() - except [SafetensorError, RuntimeError] as e: + except (SafetensorError, RuntimeError) as e: try: import os @@ -159,7 +159,7 @@ class Weights: try: bits = self.get_tensor("gptq_bits").item() groupsize = self.get_tensor("gptq_groupsize").item() - except [SafetensorError, RuntimeError] as e: + except (SafetensorError, RuntimeError) as e: try: import os