Bugfix variable reference before assignment

Prevent non-valid image paths in ImageReward.score resulting in local variable 'pil_image' referenced before assignment
THUDM · Sep 24, 2024 · c69c0f9 · c69c0f9
1 parent 8cfdfb9
commit c69c0f9
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/ImageReward/ImageReward.py b/ImageReward/ImageReward.py
@@ -112,9 +112,8 @@ def score(self, prompt, image):
         # image encode
         if isinstance(image, Image.Image):
             pil_image = image
-        elif isinstance(image, str):
-            if os.path.isfile(image):
-                pil_image = Image.open(image)
+        elif isinstance(image, str) and os.path.isfile(image):
+            pil_image = Image.open(image)
         else:
             raise TypeError(r'This image parameter type has not been supportted yet. Please pass PIL.Image or file path str.')